Python高效判断元素分属不同集合并合并的实现方法

嘿，这里有个高效的办法解决你的问题，能避开多重循环，正好匹配你的需求：

核心思路 #

我们可以先分别定位c1和c2所在的集合，只需要两次线性遍历（而非嵌套循环）就能完成定位。如果两个元素处于不同集合，就把这两个集合从原列表中移除，再将合并后的新集合加进去；如果同属一个集合或者某个元素不存在于任何集合里，就不做任何操作。

Python代码实现 #

def merge_target_sets(set_list, elem1, elem2):
    # 找到elem1所在的集合，找不到则返回None
    set_elem1 = next((s for s in set_list if elem1 in s), None)
    # 找到elem2所在的集合，找不到则返回None
    set_elem2 = next((s for s in set_list if elem2 in s), None)
    
    # 校验：两个元素都存在，且所在集合不同
    if set_elem1 and set_elem2 and set_elem1 != set_elem2:
        # 从原列表移除两个独立集合
        set_list.remove(set_elem1)
        set_list.remove(set_elem2)
        # 合并后加入列表
        set_list.append(set_elem1.union(set_elem2))
    return set_list

测试你的示例 #

original_list = [{'C', 'T'}, {'L'}, {'M'}]
result = merge_target_sets(original_list, 'C', 'M')
print(result)
# 输出示例（集合无序，顺序可能略有不同）：[{'L'}, {'C', 'T', 'M'}]

关键说明 #

1. 避免多重循环：用next()配合生成器表达式，每个元素的定位只需要一次O(n)的遍历，整体时间复杂度是O(n)，比嵌套循环的O(n²)高效得多
2. 边界处理：如果c1或c2不在任何集合里，函数会直接返回原列表，不会报错
3. 集合无序性：因为Python的集合是无序的，输出列表中集合的元素顺序可能和示例略有差异，但元素组成完全符合要求
4. 原列表修改：上面的代码会直接修改传入的原列表，如果不想改动原列表，可以先创建副本：new_list = set_list.copy()，再对new_list进行操作

内容的提问来源于stack exchange，提问作者Batman Uzamaki