140. 单词拆分 II

 

Q：

给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，在字符串中增加空格来构建一个句子，使得句子中所有的单词都在词典中。返回所有这些可能的句子。

说明：

分隔时可以重复使用字典中的单词。
你可以假设字典中没有重复的单词。
示例 1：

输入:
s = “catsanddog”
wordDict = [“cat”, “cats”, “and”, “sand”, “dog”]
输出:
[
“cats and dog”,
“cat sand dog”
]
示例 2：

输入:
s = “pineapplepenapple”
wordDict = [“apple”, “pen”, “applepen”, “pine”, “pineapple”]
输出:
[
“pine apple pen apple”,
“pineapple pen apple”,
“pine applepen apple”
]
解释: 注意你可以重复使用字典中的单词。
示例 3：

输入:
s = “catsandog”
wordDict = [“cats”, “dog”, “sand”, “and”, “cat”]
输出:
[]

A：

关键是要先用139的代码算一下给的字符串能不能被拆分为字典里的单词，不能直接返回。因为算是否能被拆分是O(N ^ 2)，而本题要求的所有序列是O(N ^ 3)，不先判断一下有一个全是’a’的用例是过不去的。。

C++:

class Solution {
public:
    vector<string> wordBreak(string s, vector<string>& wordDict) {
        vector<vector<string>> dp(s.size(),vector<string>());
        unordered_set<string> dic;
        for(auto& str:wordDict){
            dic.insert(str);
        }
        vector<bool> judge(s.size(),false);
        for(int i=0;i<s.size();++i){
            if(dic.count(s.substr(0,i+1))){judge[i]=true;}
            else{
                for(int j=0;j<=i;++j){
                    if(judge[j] and dic.count(s.substr(j+1,i-j))){
                        judge[i]=true;
                        break;
                    }
                }
            }
        }
        if(not judge.back()){return {};}
        //dp[i]表示s截止到i的可能分割出的句子
        for(int i=0;i<s.size();++i){
            for(int j=0;j<=i;++j){
                if(dic.count(s.substr(j,i-j+1))){
                    string str2=s.substr(j,i-j+1);
                    if(j==0){
                        dp[i].push_back(str2);
                        continue;
                    }
                    if(not judge[j-1]){continue;}
                    for(auto& str1:dp[j-1]){
                        dp[i].push_back(str1+" "+str2);
                    }
                }
            }
        }
        return dp[s.size()-1];
    }
};

python:

class Solution:
    def wordBreak(self, s: str, WordDict):
        l=len(s)
        if not l:
            return 0
        dic={ x:1 for x in WordDict}
        #上一题代码拿来用
        judge=[0 for i in range(l)]
        judge[0]=int(s[0] in dic)  #i==0递推基础
        for i in range(1,l):    #i==1开始遍历
            if s[:i+1] in dic:
                judge[i]=1
            else:
                for j in range(i):
                    if judge[j] and s[j+1:i+1] in dic:
                        judge[i]=1
                        break
        print(judge)
        if not judge[-1]:
            return []
        #开始算本题所求的序列
        dp=[[]for i in range(l)]  #dp[i]是一个列表，存储不同的s[1,i]闭区间的有效拆分序列
        if s[0] in dic:
            dp[0].append(s[0])  #递推起始状态
        for i in range(1,l):    #从i==1开始自底向上递推
            if s[:i+1] in dic: #s[0:i+1]本身就在字典里则加入解集
                dp[i].append(s[:i+1])
            for cut in range(i):    #进行切分，符合条件的加入解集
                if s[cut+1:i+1] in dic and judge[cut]:
                    #cut前面的子问题有解且cut之后到i的字符串在字典中
                    for x in dp[cut]:   #x为字符串
                        dp[i].append(x+' '+s[cut+1:i+1])  
                        #把dp[cut]中的每个解加上cut后的字符串，插入dp[i]的解集
        # print(dp)
        return dp[-1]