1. 程式人生 > >leetcode 720. 詞典中最長的單詞(python)

leetcode 720. 詞典中最長的單詞(python)

給出一個字串陣列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"。

注意:

 

解題思路: 這題給我的感覺很奇怪。該題用了一個set集合,判斷每個單詞的除去倒數第一個字母是否在set裡,用一個變數儲存最長的單詞。

  • 所有輸入的字串都只包含小寫字母。
  • words陣列長度範圍為[1,1000]
  • words[i]的長度範圍為[1,30]
class Solution(object):
    def longestWord(self, words):
        """
        :type words: List[str]
        :rtype: 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