【leetcode】Word Break II

Word Break II

Given a string s and a dictionary of words dict, add spaces in s to construct a sentence where each word is a valid dictionary word.

Return all such possible sentences.

For example, given
s = "catsanddog",
dict = ["cat", "cats", "and", "sand", "dog"].

A solution is ["cats and dog", "cat sand dog"].

与Word Break I类似，只不过需要把所有情况都表示出来。

首先利用Word Break I中的动态规划

dp[i]=true 代表了0,1,...,i-1可以用字典中的元素分割

同时新建一个map<int,vector<int>> slice，用于记录分割的位置

如slice[i]=j代表了0,1,...,i-1可以分割为0,1,j-1和j,j+1,...,i-1

利用这个slice，利用dfs便可以找到所有的结果

 class Solution {
 public:
     vector<string> wordBreak(string s, unordered_set<string> &dict) {

         vector<bool> dp(s.length()+);
         dp[]=true;
         map<int,vector<int> > slice;

         // dp[i]  0,1,2...i-1可以被分割
         // hash[i]=j 表示0,1,2...i-1可以分割为0,1,2,...,j-1和j,j+1,...,i
         for(int i=;i<s.length()+;i++)
         {
             for(int j=;j<i;j++)
             {
                 if(dp[j]&&dict.count(s.substr(j,i-j)))
                 {
                     dp[i]=true;

                     if(slice.find(i)!=slice.end()) slice[i].push_back(j);
                     else slice[i]=vector<int>(,j);
                 }
             }
         }

         vector<string> result;

         dfs(result,slice,s.length(),s,s);
         return result;
     }

     void dfs(vector<string> &result,map<int,vector<int>> &slice,int start,string &s,string cur)
     {
         if(start==)
         {
            cur.erase(,);
            result.push_back(cur);
            return;
         }

         vector<int> sliceV=slice[start];
         for(int i=;i<sliceV.size();i++)
         {
             string tmp=cur;
             cur.insert(sliceV[i]," ");
             dfs(result,slice,sliceV[i],s,cur);
             cur=tmp;
         }
     }

 };