leetcode_720 词典中最长的单词

113 阅读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]。

核心代码

class Solution:
    def longestWord(self, words: List[str]) -> str:
        words.sort()
        res  = set([""])
        longestWord = ""
        for word in words:
            if word[:-1] in res:
                res.add(word)
                if len(word) > len(longestWord):
                    longestWord = word
        return longestWord

image.png

解题思路:我们看到先对words中的所有单词进行排了个序,只有一个字母的单词,去掉最后一个字母后,一定在res中,所以会将单个字母的单词加入到res中,然后后续比较长的单词,减去最后一个字符,在的话,说明满足条件,看和历史最佳的单词长度关系,决定是否更新,因为我们前面已经排过序了,所以相同的长度的也是字母顺序小的在前面,满足题意。