LeetCode720_词典中最长的单词

1. 题目

给出一个字符串数组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]。

2. 题解

# 720
import collections
from typing import List


class Node:
    def __init__(self):
        self.children = collections.defaultdict(Node)
        self.isword = False


class Trie:
    def __init__(self):
        self.root = Node()

    def insert(self, word: str):
        cur = self.root
        for w in word:
            cur = cur.children[w]
        cur.isword = True

    def search(self, word: str) -> bool:
        cur = self.root
        for w in word:
            cur = cur.children.get(w)
            if cur is None or not cur.isword:
                return False
        return cur.isword


class Solution:
    def longestWord(self, words: List[str]) -> str:
        res = ""
        trie = Trie()
        for word in words:
            trie.insert(word)
        for word in words:
            if trie.search(word):
                if len(word) > len(res):
                    res = word
                elif len(word) == len(res) and word < res:
                    res = word
        return