[leetcode]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

思路也是类似的，插空的方法。但是需要把得到的结果存下来。
值得注意的是，在DP的时候我们可以选择用空间换时间也可以用时间换空间。这一题一开始提交了很多次，就是因为MLE,超过题目所要求的空间限制。
一开始代码如下：(使用vector数组b记录下每一步的结果)

class Solution {
public:
    vector<string> wordBreak(string s, unordered_set<string> &dict) {
        int n=s.size(),len,d,i,j,k;
        bool dp[n+1];
        vector <string> b[n+1];
        memset(dp,false,n+1);
        dp[0]=true;
        //b[0].push_back("");
        for(i=0;i<n+1;i++){
            for(j=0;j<i;j++){
                if(dp[j]&&dict.find(s.substr(j,i-j))!=dict.end()){
                    dp[i]=true;
                    if(j==0) b[i].push_back(s.substr(j,i-j));
                    else{
                        len = b[j].size();
                        for(k=0;k<len;k++){
                            b[i].push_back(b[j][k]+" "+s.substr(j,i-j));
                        }
                    }
                }
            }
        }
        return b[n];
    }
};

（虽然总是MLE，但是结果是正确的）结果是完全正确的。
测试代码如下：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <vector>
#include <set>
#include <string>
using namespace std;

    vector<string> wordBreak(string s, std::set<string> &dict) {
        int n=s.size(),len,d,i,j,k;
        bool dp[n+1];
        vector <string> b[n+1];
        memset(dp,false,n+1);
        dp[0]=true;
        //b[0].push_back("");
        for(i=0;i<n+1;i++){
            for(j=0;j<i;j++){
                if(dp[j]&&dict.find(s.substr(j,i-j))!=dict.end()){
                    dp[i]=true;
                    if(j==0) b[i].push_back(s.substr(j,i-j));
                    else{
                        len = b[j].size();
                        for(k=0;k<len;k++){
                            b[i].push_back(b[j][k]+" "+s.substr(j,i-j));
                        }
                    }
                }
            }
        }
        return b[n];
    }

int main(){
        set<string> dict;

        dict.insert("cat");
        dict.insert("cats");
        dict.insert("and");
        dict.insert("sand");
        dict.insert("dog");
        string s = "catsanddog";

        //cout<<s.size()<<endl;
        //cout<<b[1][0]<<endl;
        vector<string> str = wordBreak(s,dict);
        for(int i=0;i<str.size();i++) cout<<str[i]<<endl;
        system("pause");
        return 0;
}

看了这一题的tag之后，还要用回溯法。就是说需要递归。不用开那么大的vector数组b，可以节省题目的空间开销。
下面是通过的代码：

class Solution {
public:
    vector<string> wordBreakdp(string s, unordered_set<string> &dict,int cur,bool *dp){
        vector<string> ans;
        if(dp[s.size()]==false) return ans;
        else {
        for(int i=0;i<cur;i++){
        if(dp[i]&&dict.find(s.substr(i,cur-i))!=dict.end()) {
            if(i==0) ans.push_back(s.substr(i,cur-i));
            else{
            vector<string> tmp = wordBreakdp(s,dict,i,dp);
            for(int j=0;j< tmp.size();j++)   ans.push_back(tmp[j]+" "+s.substr(i,cur-i));
            }
        }
        }
        return ans;
        }
    }
    vector<string> wordBreak(string s, unordered_set<string> &dict) {
        int n=s.size(),len,d,i,j,k;
        bool dp[n+1];
        vector <string> ans;
        memset(dp,false,n+1);
        dp[0]=true;
        //b[0].push_back("");
        for(i=0;i<n+1;i++){
            for(j=0;j<i;j++){
                if(dp[j]&&dict.find(s.substr(j,i-j))!=dict.end())       dp[i]=true;
            }
        }
        return wordBreakdp(s,dict,n,dp);
    }
};