Java实现 LeetCode 720 词典中最长的单词（字典树）

小灰灰 2023-07-24 03:28 88阅读 0赞

## 720. 词典中最长的单词 ##

给出一个字符串数组words组成的一本英语词典。从中找出最长的一个单词，该单词是由words词典中其他单词逐步添加一个字母组成。若其中有多个可行的答案，则返回答案中字典序最小的单词。

若无答案，则返回空字符串。

示例 1:

输入:  
words = \[“w”,“wo”,“wor”,“worl”, “world”\]  
输出: “world”  
解释:  
单词"world"可由"w", “wo”, “wor”, 和 "worl"添加一个字母组成。  
示例 2:

输入:  
words = \[“a”, “banana”, “app”, “appl”, “ap”, “apply”, “apple”\]  
输出: “apple”  
解释:  
“apply"和"apple"都能由词典中的单词组成。但是"apple"得字典序小于"apply”。  
注意:

所有输入的字符串都只包含小写字母。  
words数组长度范围为\[1,1000\]。  
words\[i\]的长度范围为\[1,30\]。

class Solution { 
         class Node{ 
            String val;
            Node[] next;
            Node(){ 
                val=null;
                next=new Node[26];
            }
        }
        Node root;
        String res;
        //建树
        private void build(String[] words){ 
            for(String word:words)
                buildTrie(word.toCharArray(),0,root);
        }
        private void buildTrie(char[] word,int pos,Node cur){ 
            if(cur.next[word[pos]-'a']==null)
                cur.next[word[pos]-'a']=new Node();
            cur=cur.next[word[pos]-'a'];
            if(pos+1==word.length){ 
                cur.val=new String(word);
                return;
            }
            buildTrie(word,pos+1,cur);    
        }
        //查找树
        private void preorder(Node cur){ 
            if(cur.val.length()>res.length())
                res=cur.val;
            for(int i=0;i<26;i++)
                if(cur.next[i]!=null&&cur.next[i].val!=null)
                    preorder(cur.next[i]);
            // if(cur!=null){ 
            // System.out.println(cur.val);
            // for(int i=0;i<26;i++)
            // preorder(cur.next[i]);
            // }
        }
        public String longestWord(String[] words) { 
            root=new Node();
            root.val="";
            res="";
            build(words);
            preorder(root);
            return res;
        }
    }