[LeetCode]Word Break II

最新推荐文章于 2019-06-19 09:56:51 发布

原创最新推荐文章于 2019-06-19 09:56:51 发布 · 151 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#动态规划

LeetCode 专栏收录该内容

65 篇文章

订阅专栏

本文详细介绍了如何使用动态规划和回溯法解决字符串拆分问题，即给定一个字符串和一个字典，如何在字符串中插入空格以构成有效的字典单词，并返回所有可能的组合。

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"].

在I的基础上要求输出所有解，这样将所有的中间状态保存在prev二维数组中，然后再DFS即可获得所有解。

class Solution {
public:
    vector<string> wordBreak(string s, unordered_set<string> &dict) {
        vector<bool> dp(s.length() + 1, false);
        dp[0] = true;
        vector<vector<int>> prev(s.length() + 1);
        for (int i = 1; i <= s.length(); i++)
        {
            for (int j = i - 1; j >= 0; j--)
            {
                if (dp[j] && dict.count(s.substr(j, i - j)))
                {
                    dp[i] = true;
                    prev[i].push_back(j);
                }
            }
        }
        vector<string> result;
        searchSolution(result, prev, s, "", s.length());
        return result;
    }
    
    void searchSolution(vector<string> &sol, vector<vector<int>> &prev, string &s, string line, int k)
    {
        if (k == 0)
        {
            if(line.empty()) return;
            line.resize(line.length() - 1); // remove last space character
            sol.push_back(line);
            return;
        }
        for (auto j : prev[k])
            searchSolution(sol, prev, s, s.substr(j, k - j) + " " + line, j);
    }
};