leetcode[140] Word Break II

这题是上一题的加强版，这里需要返回的是所有可能的分割词。例如：

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

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

先用dp求得每个起点到终点是否符合word break 1中的条件，然后利用DFS来递归求解答案。需要注意的是，为了递归的方便，要注意dp要多一个dp[s.size() + 1]长度

class Solution {
public:
    //140
    void dfs140(string s, unordered_set<string> &dict, vector<string> &ans, vector<string> &vec, bool dp[], int start)
    {
        if (start == s.size())
        {
            if (vec.size() > 0)
            {
                string str = vec[0];
                for (int i = 1; i < vec.size(); i++)
                {
                    str = str + " " + vec[i];
                }
                ans.push_back(str);
            }
            return ;
        }
        for (int i = 1; i <= s.size() - start; i++)
        {
            if (dict.count(s.substr(start, i)) && dp[start + i])
            {
                vec.push_back(s.substr(start, i));
                dfs140(s, dict, ans, vec, dp, start + i);
                vec.pop_back();
            }
        }
    }
    vector<string> wordBreak(string s, unordered_set<string> &dict)
    {
        bool dp[s.size() + 1]; // 不同于上一题，这里需要多一位，最后一位为真，为了递归的时候方便
        memset(dp, 0, sizeof(dp));
        for (int i = 0; i < s.size(); i++)
        {
            if (dict.count(s.substr(i, s.size() - i)))
                dp[i] = 1;
        }
        for (int i = s.size() - 1; i < s.size(); i--)
        {
            if (dp[i] == false)
            {
                for (int j = i; j < s.size(); j++)
                {
                    if (dict.count(s.substr(i, j - i + 1)) && dp[j + 1])
                        dp[i] = true;
                    if (dp[i])
                        break;
                }
            }
        }
        dp[s.size()] = 1;
        vector<string> ans, tmp;
        dfs140(s, dict, ans, tmp, dp, 0);
        return ans;
    }
};

这题的思路很像leetcode Palindrome Partitioning II