LintCode 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.

样例

Gieve s = lintcode,
dict = ["de", "ding", "co", "code", "lint"].

A solution is ["lint code", "lint co de"].

 解法

之前做了一道单词切分的题目，直接深度优先搜索，超时后改成非递归形式的就过了，所以认为这道题就是考深度优先搜索的。面对这个加强版的题也如法炮制，发现非递归形式也会超时，上网查了下才知道这题的核心是一个判断字符串是否能被划分的dp算法。

直接深度优先太慢，所以要先判断一下排除掉不可能划分的情况。dp[i] 表示从 i 开始到最后的子字符串可以被切分， len 表示字符串的长度。则有 dp[len] = true，当从 i 到 j 的字符串在字典中且 dp[j + 1] = true 时 dp[i] = true。有了这个表再进行深度优先搜索找出所有可能的划分，不可能划分的就不用算了。

vector<string> wordBreak(string s, unordered_set<string> &wordDict) {        // Write your code here        vector<bool> dp(s.size() + 1, true);        string dps;        for(int i = s.size() - 1; i >= 0; --i) {            dps.clear();            dp[i] = false;            for(int j = i; j < s.size(); ++j) {                dps.append(1, s[j]);                if(dp[j + 1] && wordDict.find(dps) != wordDict.end()) {                    dp[i] = true; break;                }            }        }                vector<string> r;        vector<string> buf;        sub(s, wordDict, 0, buf, r, dp);        return r;    }            void sub(string& s, unordered_set<string> &dict, int cur, vector<string>& buf, vector<string>& r, vector<bool>& dp) {        string t;        if(cur >= s.size()) {            t = buf[0];            for(int i = 1; i < buf.size(); ++i) {                if(i > 0) t.append(1, ' ');                t.append(buf[i]);            }                r.push_back(t);            return;        }        if(!dp[cur]) return;        for(int i = cur; i < s.size(); ++i) {            t.append(1, s[i]);            if(dict.find(t) != dict.end()) {                buf.push_back(t);                sub(s, dict, i + 1, buf, r, dp);                buf.pop_back();            }        }    }