如何存储支持双向快速查询的等价名称对应关系?
你需要的这种等价关系双向查询场景,有两种成熟的实现方案,没有你举例的{'Adam':'Lawrence':'Albert'}这类链式键的原生数据结构:
方案1:双向字典映射(适合数据量小、等价集合固定的场景)
核心逻辑是给每个名称绑定唯一的组主标识,再用单独的字典存储组内所有等价名,查询时两步即可拿到结果,复杂度O(1):
- 维护两个字典:
alias_to_root:存储所有名称对应的组主名,比如{'Adam':'Adam', 'Lawrence':'Adam', 'Albert':'Adam', 'Heintz':'Erwin'}root_to_equivalents:存储组主名对应的所有等价名列表,比如{'Adam': ['Adam', 'Lawrence', 'Albert'], 'Erwin': ['Erwin', 'Heintz', 'Edwin', 'Bro']}
- 查询逻辑:输入任意名称时,先通过
alias_to_root拿到对应组主名,再通过root_to_equivalents拿到该组全部等价名即可。
下面是Python实现示例:
# 原始标注数据 raw = "Adam (Lawrence, Albert), Erwin (Heintz, Edwin, Bro)" alias_to_root = {} root_to_equivalents = {} # 解析数据构建映射 for group_str in raw.split(", "): if "(" not in group_str: continue root, alias_part = group_str.split(" (") alias_list = [root] + alias_part.rstrip(")").split(", ") root_to_equivalents[root] = alias_list for name in alias_list: alias_to_root[name] = root # 通用查询方法 def get_all_equivalents(name: str) -> list: root = alias_to_root.get(name) return root_to_equivalents.get(root, []) # 测试 print(get_all_equivalents("Lawrence")) # 输出:['Adam', 'Lawrence', 'Albert'] print(get_all_equivalents("Erwin")) # 输出:['Erwin', 'Heintz', 'Edwin', 'Bro']
方案2:并查集(Disjoint Set Union, DSU,适合数据量大、需要频繁合并等价组的场景)
如果后续会新增等价关系、需要合并不同的等价组,用并查集更合适,它的查找、合并操作平均复杂度接近O(1),只需要给每个并查集的根节点绑定存储该组所有等价名的列表,查询时找到任意名称对应的根节点,就能直接拿到全部等价名。
内容的提问来源于stack exchange,提问作者guyguyguy12345
相关产品推荐
相关产品推荐

