单词拆分

单词拆分

给你一个字符串 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

提示：

• 1 <= s.length <= 300
• 1 <= wordDict.length <= 1000
• 1 <= wordDict[i].length <= 20
• s 和 wordDict[i] 仅有小写英文字母组成
• wordDict 中的所有字符串 互不相同

代码：

//DP
class Solution {
public:
    bool wordBreak(string s, vector<string>& wordDict) {
        auto wordDictSet = unordered_set <string> ();
        for (auto word: wordDict) {
            wordDictSet.insert(word);
        }

        auto dp = vector <bool> (s.size() + 1);
        dp[0] = true;
        for (int i = 1; i <= s.size(); ++i) {
            for (int j = 0; j < i; ++j) {
                if (dp[j] && wordDictSet.find(s.substr(j, i - j)) != wordDictSet.end()) {
                    dp[i] = true;
                    break;
                }
            }
        }

        return dp[s.size()];
    }
};

//字典树 + DP
class Solution {
public:
    bool wordBreak(string s, vector<string>& wordDict) {
        dp = vector<int>(s.size(),0);
        root = new TrieNode();
        auto node = root;
        for(auto word: wordDict){
            node = root;
            for(auto c:word){
                if(node->next.count(c) == 0){
                    node->next[c] = new TrieNode();
                }
                node =node->next[c];
            }
            node->flag = true;
        }
        return func(s,0,s.size());
    }

private:
    struct TrieNode{
        bool flag;
        map<char,TrieNode*> next;
        TrieNode():flag(false){}
    };
    TrieNode *root;
    vector<int> dp;
    bool func(string& s,int start,int end){
        if(start == end) return true;
        if(dp[start]!=0) return dp[start]>0;
        auto node = root;
        for(int i = start;i<end;++i){
            if(node->next.count(s[i]) == 0)break;
            node = node->next[s[i]];
            if(node->flag&&func(s,i+1,end)){
                dp[start] = 1;
                return true;
            }
        }
        dp[start] = -1;
        return false;
    }
};