给定一个非空字符串 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
思路:动态规划来解决,设flag[i]表示s的前i个字符是否可以进行单词拆分(flag[0]=true),将整个字符串分成两部分:前j个字符,s.substr(j, i-j)。那么状态转移方程为:
flag[i]=true,if flag[j]=true && s.substr(j, i-j)在字典里。
找j的过程,从后往前找,速度快。
class Solution {
public:
bool wordBreak(string s, vector<string>& wordDict) {
if(wordDict.empty()) return false;
unordered_set<string>dic(wordDict.begin(), wordDict.end());
vector<bool>flag(s.size()+1, false);
flag[0]=true;
for(int i=1; i<flag.size(); ++i){
for(int j=i-1; j>=0; --j){//从后往前找,找得快
if(flag[j] && dic.count(s.substr(j, i-j))){
flag[i]=true;
break;
}
}
}
return flag.back();
}
};