Given a string s and a dictionary of words dict, add spaces in s to construct a sentence where each word is a valid dictionary word.
Return all such possible sentences.
For example, given
s = "catsanddog"
,
dict = ["cat", "cats", "and", "sand", "dog"]
.
A solution is ["cats and dog", "cat sand dog"]
.
直接根据word breakI 的f[][]数组去build结果
class Solution {
public:
vector<string> wordBreak(string s, unordered_set<string> &dict) {
// Note: The Solution object is instantiated only once and is reused by each test case.
vector<string> ret;
if(s.length() == 0) return ret;
int len = s.length();
int ** f = new int*[len];
for(int i = 0; i < len; i++) {
f[i] = new int[len];
for(int j = 0; j < len; j++) {
string tmp = s.substr(i,j-i+1);
if(dict.find(tmp) != dict.end()) f[i][j] = 1;
else f[i][j] = 0;
}
}
for(int i = 0; i < len; i++) {
for(int j = i; j < len; j++) {
for(int k = j+1; k < len; k++) {
if(f[i][j] == 1 && f[j+1][k] == 1) {
f[i][k] = 1;
}
}
}
}
vector<string> tmp;
build(s,f,0,tmp,ret,dict);
return ret;
}
void build(string &s, int **f, int start,vector<string> tmp, vector<string> &ret,unordered_set<string> &dict) {
int len = s.length();
if(start > len) return ;
if(start == len) {
string t = "";
for(int i = 0; i < tmp.size(); i++) {
if(i == 0) t += tmp[i];
else t += " " + tmp[i];
}
ret.push_back(t);
return;
}
for(int i = start; i < len; i++) {
string t = s.substr(start,i-start+1);
if(dict.find(t) != dict.end()) {
if(i == len-1) {
tmp.push_back(t);
build(s,f,i+1,tmp,ret,dict);
tmp.pop_back();
}else {
if(f[i+1][len-1]) {
tmp.push_back(t);
build(s,f,i+1,tmp,ret,dict);
tmp.pop_back();
}
}
}
}
}
};