LeetCode 720. 词典中最长的单词

思路

先将所有单词存入字典树。对于每个单词，在字典树中检查它的全部前缀是否存在。

代码实现

class Solution {

    class Trie {
    public:
        bool isWord = false;
        Trie* next[26] = {NULL};

        Trie(){}
        void insert(const string& word) {
            Trie* t = this;
            for(int i = 0; i < word.length(); ++i){
                if(t->next[word[i]-'a'] == NULL) {
                    t->next[word[i]-'a'] = new Trie();
                }

                t = t->next[word[i]-'a'];
            }

            t->isWord = true;
        }

        //检查word单词的所有前缀是否都在字典树中
        bool check(const string& word) {
            Trie* t = this;
            for(int i = 0; i < word.length(); ++i){
                if(t->next[word[i]-'a']->isWord == false) {
                    return false;
                }
                t = t->next[word[i]-'a'];
            }
            return true; 
        }
        
    };
    
public:
    string longestWord(vector<string>& words) {
        Trie* t = new Trie();
        for(int i = 0; i < words.size(); ++i) {
            t->insert(words[i]);
        }

        string ans = "";
        for(int i = 0; i < words.size(); ++i) {
            if(t->check(words[i]) == true) {
                if(words[i].length() > ans.length())
                    ans = words[i];
                else if(words[i].length() == ans.length() && words[i] < ans)
                    ans = words[i];
            }
        }

        return ans;
    }
};

复杂度分析

时间复杂度：O(所有单词长度之和)，即创建前缀树占主要。

空间复杂度：O(所有单词长度之和)，创建前缀树耗费的空间。