139. 单词拆分

给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，判定 s 是否可以被空格拆分为一个或多个在字典中出现的单词。

说明：

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

输入: s = "leetcode", wordDict = ["leet", "code"]
输出: true
解释: 返回 true 因为 "leetcode" 可以被拆分成 "leet code"。
示例 2：

输入: s = "applepenapple", wordDict = ["apple", "pen"]
输出: true
解释: 返回 true 因为 "applepenapple" 可以被拆分成 "apple pen apple"。
     注意你可以重复使用字典中的单词。
示例 3：

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

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

方法一：动态规划

设$dp[i]$表示在i之前是否在字典中存在。

代码

bool wordBreak(string s, vector<string>& wordDict) {
    vector<bool> dp(s.size()+1, false);
    unordered_set<string> m(wordDict.begin(), wordDict.end());
    dp[0] = true;
    for (int i = 1; i <= s.size(); ++i){
        for (int j = 0; j < i; ++j){
            if (dp[j] && m.find(s.substr(j, i-j)) != m.end()){
                dp[i] = true;
                break;
            }
        }
    }
    return dp[s.size()];
}

进一步优化

其实不是j每一次都要从0开始迭代，因为字典里的单词是有长度限制的，只要$i-j>最大单词长度$即可。

class Solution {
public:
    bool wordBreak(string s, vector<string>& wordDict) {
        unordered_set<string> table(wordDict.begin(), wordDict.end());
        int N = s.size();
        vector<bool> dp(N+1, false);
        dp[0] = true;
        int maxWordLength = 0;
        for (int i = 0; i < wordDict.size(); ++i){
            maxWordLength = std::max(maxWordLength, (int)wordDict[i].size());
        }

        for(int i=1; i<=N; ++i)
        {
            for(int j=std::max(i-maxWordLength, 0); j<i; ++j)
            {
                if(dp[j]&&table.find(s.substr(j, i-j))!=table.end()){
                    dp[i] = true;
                    break;
                }
            }

        }
        return dp[N];
    }
};