给你一个字符串 s
和一个字符串列表 wordDict
作为字典。如果可以利用字典中出现的一个或多个单词拼接出 s
则返回 true
。
**注意:**不要求字典中出现的单词全部都使用,并且字典中的单词可以重复使用。
cpp
class Solution {
public:
bool wordBreak(string s, vector<string>& wordDict) {
unordered_set<string> wordSet(wordDict.begin(), wordDict.end());
int n = s.size();
vector<bool> dp(n + 1, false);
dp[0] = true;
int maxLen = 0;
for (const string& word : wordDict) {
maxLen = max(maxLen, (int)word.length());
}
for (int i = 1; i <= n; ++i) {
int start = max(0, i - maxLen);
for (int j = start; j < i; ++j) {
if (dp[j] && wordSet.count(s.substr(j, i - j))) {
dp[i] = true;
break;
}
}
}
return dp[n];
}
};
状态转换核心逻辑是,如果dp[j]为true,并且(j,i-j)这一段在字典中,那么dp[i]也为真
定义了一个start优化效率,逻辑是,如果此时i大于最大字符串长度,那就没必要从0开始遍历