【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()+1);
        dp[0]=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=1;i<s.length()+1;i++)
        {
            for(int j=0;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>(1,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==0)
        {
           cur.erase(0,1);
           result.push_back(cur);
           return;
        }
       
        vector<int> sliceV=slice[start];
        for(int i=0;i<sliceV.size();i++)
        {
            string tmp=cur;
            cur.insert(sliceV[i]," ");
            dfs(result,slice,sliceV[i],s,cur);
            cur=tmp;
        }
    }
   
};