考察点:dp,看是否一个string可以由其他string组成的变种;
思路:首先要会一个string可以由其他string组成这道题。然后就可以依次按照string的length长短来判断是否满足上一道字问题。注意一点的是再判断时在第二个for循环下j应该按照从0到i的遍历一次而不是按照string集合去遍历,因为那样会超时。
C++ 代码:
class Solution {
public:
vector<string> findAllConcatenatedWordsInADict(vector<string>& words) {
vector<string> ret;
if (words.size() == 1)
return ret;
sort(words.begin(), words.end(), [](string &a, string &b){
return a.size() < b.size();
});
set<string> set1;
for (auto item : words) {
if (isValid(set1, item)) {
ret.push_back(item);
}
set1.insert(item);
}
return ret;
}
bool isValid(set<string> &set1, string &item) {
if (set1.empty())
return false;
int len = item.size();
bool dp[len+1];
memset(dp, 0, sizeof(dp));
dp[0] = true;
for (int i=1; i<=len; i++) {
for (int j = 0; j<i; j++) {
if (dp[j] && set1.find(item.substr(j, i-j)) != set1.end()) {
dp[i] = true;
break;
}
}
}
return dp[len];
}
};