题目链接:https://leetcode-cn.com/problems/word-break/
题目描述
给定一个非空字符串 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
示例4:
输入: s = "aaaaaaa", wordDict = ["aaaa", "aaa"]
输出: true
思路
该题考察点是动态规划。刚开始想用暴力的方法顺序遍历字符串,查看是否和字典匹配。但是示例4中可以看出,如果一开始匹配了"aaa"这个字符串两次,则会剩下一个字符a,无法得到匹配。所以在匹配的过程中是有一个回溯的过程的,如果剩余字符不匹配,则需要回溯上一个匹配单词。
但是在该题中如果用回溯来做,需要记下每次匹配的单词的位置,在回溯后匹配后续单词,比较繁琐。
考虑用动态规划来做。
- dp[i]表示字符串s的前i个字符能否拆分成wordDict
- 如果d[j]=true && (i-j)组成的字符串在字典中,则d[i]=true
- 对于每一个i,都需要一个内循环去对于所有j<i检查,是否满足上述条件
代码
class Solution {
public:
// 动态规划。dp[i]表示字符串s的前i个字符能否拆分成wordDict
// 如果d[j]=true && (i-j)组成的字符串在字典中,则d[i]=true
bool wordBreak(string s, vector<string>& wordDict) {
vector<bool> dp(s.size()+1, false);
unordered_set<string> m(wordDict.begin(), wordDict.end());
dp[0] = true;
for (int i = 1; i <= s.size(); ++i) {
for (int j = 0; j < i; ++j) {
if(dp[j] && (m.find(s.substr(j,i-j))!=m.end())){
dp[i] = true;
break;
}
}
}
return dp[s.size()]; // 前n个字符能否拆分
}
};