题目:
题解:
思路1:前缀树
- 最长单词的每个字符在前缀树中对应的节点都能表示一个字符串,还有一点需要注意的是在最长单词的长度相等时,我们需要选择字典序小的单词。然后就是两次遍历字符串数组,第一次遍历是建立前缀树,第二次遍历是寻找最长单词了。
思路2:用 set 来判断每个的单词的所有前缀都是否存在。
- 所有前缀存在的话,然后选择一个长度最长的,或者遇到长度相同的字符串,选择字典序更小的那个。
思路3:排序
- 排序:先将 words 中的单词进行排序,对于长度不同的单词,按长度由小到大进行排序;对于长度相同的元素,按字典序由大到小进行排序。这样来确保遍历到每个单词时,比它短的前缀都遍历过了,而已每遇到符合要求的单词都是长度最长且字典序最小的单词了。
- 遍历:由于已经排序过了,则当前单词的所有前缀只有在set中,才能进行更新res,而且我们只用判断当前单词的最长前缀即可。
代码如下:
// 思路1:前缀树
class Trie{
private:
bool is_string=false;
Trie *next[26]={nullptr};
public:
Trie(){}
void insert(string word){// 插入单词
Trie *root=this;
for(const auto& w:word){
if(root->next[w-'a']==nullptr)root->next[w-'a']=new Trie();
root=root->next[w-'a'];
}
root->is_string=true;
}
bool search(string word){// 判断该单词的所有前缀是否由其他所有单词构成
Trie *root=this;
for(const auto& w:word){
// 该前缀不是由其他单词逐步组成的,查找失败
if(!root->next[w-'a']->is_string)return false;
root=root->next[w-'a'];
}
return true;
}
};
class Solution {
public:
string longestWord(vector<string>& words) {
Trie *root=new Trie();
for(const auto& w:words)root->insert(w);
string res;
for(const auto& w:words)
if(root->search(w))
{
// 长度不一样,则更新为更长的
if(res.size()<w.size())res=w;
// 长度一样,则更新为字典序更小的
else if(res.size()==w.size())res=min(res,w);
}
return res;
}
};
// 思路2:用 set 来判断是否该单词的所有前缀都存在
class Solution {
public:
string longestWord(vector<string>& words) {
set<string> s(words.begin(),words.end());
string res;
for(const auto& w:words){
int i,n=w.size();
// 判断该单词的前缀是否都存在于words中
for(i=0;i<n;++i){
if(!s.count(w.substr(0,i+1)))break;
}
// 该单词的前缀都存在,则判断是否进行更新
if(i==n){
// 长度不一样,则更新为更长的
if(res.size()<w.size())res=w;
// 长度一样,则更新为字典序更小的
else if(res.size()==w.size())res=min(res,w);
}
}
return res;
}
};
// 思路3:排序
class Solution {
public:
string longestWord(vector<string>& words) {
// 将数组words中的单词排序,对于长度不同的单词,按长度由小到大进行排序;对于长度相同的元素,按字典序由大到小进行排序
// 这样来确保遍历到每个单词时,比它短的前缀都遍历过了,而已每遇到符合要求的单词都是长度最长且字典序最小的单词了
sort(words.begin(),words.end(),[](const auto& a,const auto& b){
return a.size()!=b.size()?a.size()<b.size():a>b;
});
string res;
// 首先加入空字符串
unordered_set<string> dict={""};
for(const auto& w:words){
// 由于已经排序过了,则当前单词的所有前缀只有在set中,才能进行更新res,而且我们只用判断当前单词的最长前缀即可
if(dict.count(w.substr(0,w.size()-1))){
res=w;
dict.insert(w);
}
}
return res;
}
};