题目
给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict,在字符串中增加空格来构建一个句子,使得句子中所有的单词都在词典中。返回所有这些可能的句子。
说明:
分隔时可以重复使用字典中的单词。
你可以假设字典中没有重复的单词。
解题思路
这道题明显是个搜索的题目。
但问题在于如何进行搜索。每个位置可能变成除自身以外的25个字母。直接搜会很暴力。
所以我们想了一种利用dp+字典树的方法来先将单词进行划分。
若dp[i]=j,表示字符串从下标i开始,长度为j-i的字符串存在在题目给定的字典里面。
如何查找呢?最好的方法就是将字典里面的单词建立一个字典树,然后在树上查找就好了。
在建立dp数组之后,需要从末尾开始DFS搜索结果就好了。
代码如下:
//字典树的模板
class Tire{
public:
bool isLeaf;
Tire* next[26];
Tire(){
isLeaf = false;
memset(next,0,sizeof(next));
}
void Insert(string s){
Tire *node = this;
for(char c : s){
if(node->next[c - 'a'] == NULL){
node->next[c - 'a'] = new Tire();
}
node = node->next[c - 'a'];
}
node->isLeaf = true;
}
};
class Solution {
public:
void DFS(string s,vector<string>& ans,vector<string>& tWord,vector<vector<int>>dp,int d){
if(d == 0){
string str ="";
for(auto it = tWord.rbegin();it != tWord.rend();it++){
str += *it+" ";
}
str.pop_back();//最后末尾多余一个" "
ans.push_back(str);
return;
}
for(int i : dp[d]){
tWord.push_back(s.substr(i,d - i));
DFS(s,ans,tWord,dp,i);
tWord.pop_back();
}
}
vector<string> wordBreak(string s, vector<string>& wordDict){
Tire* root = new Tire();
for(string word : wordDict) root->Insert(word);
int l = s.size();
vector<vector<int> > dp(l + 1);
dp[0].push_back(-1);
for(int i = 0;i < l;i++){
if(dp[i].empty()) continue;
int j = i;
Tire* cur = root;
while(j < l && cur->next[s[j] - 'a']){
cur = cur->next[s[j++] - 'a'];
if(cur->isLeaf){
dp[j].push_back(i);
//表示以j结尾的前缀单词的开始下标在i
}
}
}
//得到dp数组之后,从最后一个字母开始向前DFS,最终可以搜索到所有的单词。
vector<string> ans;
vector<string> tWord;
DFS(s,ans,tWord,dp,l);
return ans;
}
};