单词拆分
给你一个字符串 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;
}
};