Phone List（字典树）-白红宇

Phone List（字典树）

阅读量：621 次

发布时间：2019-03-13

本文共 1427 字，大约阅读时间需要 4 分钟。

为了解决这个问题，我们需要判断给定的电话号码列表是否一致，即没有电话号码是另一个号码的前缀。我们可以使用前缀树（trie）结构来高效地解决这个问题。

方法思路

前缀树结构：使用前缀树来存储电话号码。每个节点代表一个数字，子节点表示接下来的数字。每个节点还记录是否是一个完整的号码结尾。

插入电话号码：对于每个电话号码，从根节点开始逐步遍历每一位数字。如果在某个节点已经存在，说明这个号码是前缀，返回错误。

检查冲突：在插入每个号码时，检查是否有前缀冲突。如果发现冲突，立即返回错误结果。

解决代码

class TrieNode:
    def __init__(self):
        self.children = {}
        self.is_end = False
def is_consistent(numbers):
    root = TrieNode()
    for num in numbers:
        current = root
        for c in num:
            digit = int(c)
            if digit not in current.children:
                current.children[digit] = TrieNode()
            current = current.children[digit]
            if current.is_end:
                return False
        current.is_end = True
    return True
def main():
    import sys
    input = sys.stdin.read().split()
    ptr = 0
    t = int(input[ptr])
    ptr += 1
    for _ in range(t):
        n = int(input[ptr])
        ptr +=1
        numbers = []
        for _ in range(n):
            num = input[ptr].strip()
            ptr +=1
            numbers.append(num)
        if is_consistent(numbers):
            print("YES")
        else:
            print("NO")
if __name__ == "__main__":
    main()

代码解释

TrieNode类：每个节点包含子节点和一个标记来记录是否是叶子节点（即是否是一个完整的号码结尾）。

is_consistent函数：使用前缀树插入每个号码，检查是否有前缀冲突。如果发现冲突，返回False，否则返回True。

main函数：读取输入数据，处理每个测试用例，调用is_consistent函数检查一致性，并输出结果。

这种方法的时间复杂度是O(n * 10)，其中n是电话号码的数量，10是电话号码的最大长度，能够高效处理题目中的输入规模。

转载地址：http://xueaz.baihongyu.com/

你可能感兴趣的文章