题目描述
给你一个字符串 s 和一个字符串列表 wordDict 作为字典。如果可以利用字典中出现的一个或多个单词拼接出 s 则返回 true。
注意:不要求字典中出现的单词全部都使用,并且字典中的单词可以重复使用。
示例 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 中的所有字符串 互不相同
思路
动态规划模板,使用哈希记录每个字符串,每次看第i个位置减去这些字符串后的位置是否为true即可,详细见代码。
代码
cpp
class Solution {
public:
bool wordBreak(string s, vector<string>& wordDict) {
// 记录所有字符串
unordered_set<string>ust;
// 记录最长的字符串长度
int maxs = 0;
for(const string &word : wordDict)
{
ust.insert(word);
maxs = max(maxs, (int)word.length());
}
int n = s.length();
vector<bool>dp(n, false);
for(int i = 0; i < n; ++i)
{
int m = min(maxs, i + 1);
for(int j = m; j; --j)
{
int k = i - j + 1;
if((k == 0 || dp[k - 1]) && ust.find(s.substr(k, j)) != ust.end())
{
dp[i] = true;
break;
}
}
}
return dp[n - 1];
}
};