LeetCode 472 连接词

原创于 2021-12-29 16:30:22 发布 · 87 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #深度优先 #算法

leetcode 专栏收录该内容

64 篇文章

订阅专栏

该博客介绍了一种使用字典树（Trie）数据结构来查找给定单词列表中所有可能的连接词的方法。首先，通过排序单词按长度升序，然后遍历每个单词，利用DFS搜索字典树判断是否为连接词。若单词存在，即为连接词，否则将其插入字典树。最后返回所有连接词。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Trie:
    def __init__(self):
        self.children = [None] * 26
        self.isEnd = False

    # 根据插入的word构造字典树
    def insert(self, word):
        node = self
        for ch in word:
            ch = ord(ch) - ord("a")
            if not node.children[ch]:
                node.children[ch] = Trie()
            node = node.children[ch]
        node.isEnd = True

    # 遍历word是否满足字典树
    def dfs(self, word, start):
        if start == len(word):
            return True
        node = self
        for i in range(start, len(word)):
            node = node.children[ord(word[i]) - ord("a")]
            if node is None:
                return False
            if node.isEnd and self.dfs(word, i + 1):
                return True
        return False


def findAllConcatenatedWordsInADict(words):
    words.sort(key=len)
    ans = []
    root = Trie()
    for word in words:
        if word == "":
            continue
        # 如果当前单词存在于字典树中，则是连接词
        if root.dfs(word, 0):
            ans.append(word)
        # 否则加入到字典树中
        else:
            root.insert(word)
    return ans


if __name__ == '__main__':
    findAllConcatenatedWordsInADict(
        ["cat", "cats", "catsdogcats", "dog", "dogcatsdog", "hippopotamuses", "rat", "ratcatdogcat"])

把每一个words中的元word构建一个大的字典树，判断两次满足元单词的word加入到最终的结果中