LeetCode 140——单词拆分 II（动态规划+DFS）

一、题目介绍

给定一个非空字符串 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"]
输出:
[]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/word-break-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、解题思路

（1）通过动态规划的方法，记录前i个字符串是否可以用词典中的单词拆分，即确定状态。

（2）通过深度优先搜索的方法，遍历所有情况的组合。

三、解题代码

class Solution {
public:
    int n;
    vector<string> res;
    vector<bool> dp; //代表前i个字符是否可以被词典中的单词拆分
    vector<string> wordBreak(string s, vector<string>& wordDict) {
        n = s.size();
        dp.assign(n+1, false);
        dp[0] = true; //初始状态
        set<string> st(wordDict.begin(), wordDict.end());
        int maxlen = 0;
        for(int i = 0; i < wordDict.size(); ++i)
        {
            maxlen = max(maxlen, (int)wordDict[i].size());
        }
        //状态转移
        for(int i = 1; i <= n; ++i)
        {
            for(int j = max(0, i-maxlen); j < i; ++j)
            {
                if(dp[j] && st.count(s.substr(j, i-j)) == 1)
                {
                    dp[i] = true;
                    break;
                }
            }
        }
        
        if(dp[n])
        {
            deque<string> que;
            dfs(res, n, wordDict, s, que, st);    
        }
        return res;        
    }

    //深度优先搜索所有情况
void dfs(vector<string>& res, int nsize, vector<string>& wordDict, string s, deque<string>& que, set<string>& st)
    {
            if(nsize == 0)
            {
                string stemp;
                for(auto item : que)
                {
                    stemp += item + " ";
                }
                if(!stemp.empty())
                {
                    res.push_back(string(stemp.begin(), stemp.end()-1));
                }
                return;
            }
            for(int i = 0; i < nsize; ++i)
            {
                if(dp[i]) //首先前i个字符能够被词典中的单词拆分
                {
                    string str = s.substr(i, nsize-i); //先找到最后一个可以拆分的单词
                    if(st.count(str) == 1) //如果剩下的字符组成的单词在词典中可以查到
                    {
                        que.push_front(str);
                        dfs(res, i, wordDict, s, que, st);
                        que.pop_front(); //恢复现场
                    }
                }
            }
    }
};

四、解题结果