题目:
给定一个字符串 s 和一个字符串字典 wordDict ,在字符串 s 中增加空格来构建一个句子,使得句子中所有的单词都在词典中。以任意顺序 返回所有这些可能的句子。
题解:
很简单的思路,就是一个深搜,加剪枝
如果字符串前 n 个能和字符串字典中的匹配上,则继续匹配后面的
判断是否和字符串字典匹配上,可以把字符串字典中的放入 set 集合,然后 count() 判断即可
可以加上记忆化搜索,效率更高
比如从第 i 个开始匹配,可以匹配到的数量,避免重复搜索
代码如下:
class Solution {
public:
vector<string> res;
void solve(string str, int n, string r, unordered_set<string> temp) {
if(temp.count(str.substr(0, n)) != 0) {
if(n == str.size()) {
r = r + str;
res.push_back(r);
return;
}
r = r + str.substr(0, n) + ' ';
string s = str.substr(n);
for(int i = 1; i <= s.size(); i++) {
solve(s, i, r, temp);
}
}
else {
return;
}
}
vector<string> wordBreak(string s, vector<string>& wordDict) {
unordered_set<string> temp(wordDict.begin(), wordDict.end());
for(int i = 1; i <= s.size(); i++) {
solve(s, i, "", temp);
}
return res;
}
};