139. 单词拆分. dp

Posted 2020-12-12 xgbt

给定一个非空字符串 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
public:
    bool wordBreak(string s, vector<string>& wordDict) {
        int n = s.size();
        //dp[i]表示字符串s的前i个字符能否拆分成wordDict。
        vector <bool> dp (n + 1, false);
        unordered_set <string> m(wordDict.begin(), wordDict.end());

        dp[0] = true;
        //遍历起始位置
        for (int i = 1; i <= n; 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[n];
    }
};

优化
每次并不需要从s[0]开始搜索。因为wordDict中的字符串长度是有限的。只需要从i-maxWordLength开始搜索就可以了。

class Solution {
public:
    bool wordBreak(string s, vector<string>& wordDict) {
        int n = s.size();
        //dp[i]表示字符串s的前i个字符能否拆分成wordDict。
        vector <bool> dp (n + 1, false);
        unordered_set <string> m(wordDict.begin(), wordDict.end());

        int maxLength = 0;
        for (auto word : wordDict){
            maxLength = max(maxLength, (int)word.size());
        }

        dp[0] = true;
        //遍历起始位置
        for (int i = 1; i <= n; i++){
            //遍历长度
            for (int j = max(i - maxLength, 0); j < i; j++){
                if (dp[j] && m.find(s.substr(j, i - j)) != m.end()){
                    dp[i] = true;
                    break;
                }
            }
        }

        return dp[n];
    }
};