Python多列表合并时重叠跟踪与互斥拆分的实现方法

我们需要对N个（N≥2）列表进行合并处理，实现三个核心目标：

• 最终输出的所有Final列表元素完全互斥，无重复
• 多列表重叠的元素尽可能均匀分配到各个列表中
• 每个列表对应生成Skips记录，存储所有分配给其他列表的重叠元素

你当前的双列表处理方案逻辑清晰，核心是通过交集识别重叠元素，再按索引奇偶性轮询分配，代码如下：

Alpha = ['1', '2', '3', '4', '6', '8']
Bravo = ['1', '2', '3', '5', '7', '9']

Overlap = set(Alpha).intersection(Bravo) # ['1','2','3']

AlphaFinal = [record for record in Alpha if record not in Overlap] # ['4','6','8']
BravoFinal = [record for record in Bravo if record not in Overlap] # ['5', '7', '9']

for Index, Value in enumerate(Overlap):
    if int(Index) % 2 == 0:
        AlphaFinal.append(Value) # ['2']
        BravoSkips.append(Value) # ['2']
    else:
        BravoFinal.append(Value) # ['1', '3']
        AlphaSkips.append(Value) # ['1', '3']

# AlphaFinal ends as ['4','6','8', '2']
# AlphaSkips ends as ['1', '3']
# BravoFinal ends as ['5','7','9','1','3']
# BravoSkips ends as ['2']

实现思路 #

1. 先统计所有元素的出现频次，筛选出出现次数≥2的重叠元素集合
2. 每个列表先过滤掉所有重叠元素，得到初始的唯一元素集合作为Final列表初始值，同时初始化对应空的Skips列表
3. 遍历所有重叠元素，按索引对列表总数取模的结果，分配给对应索引的列表，同时将该元素添加到其他所有列表的Skips记录中，保证分配的均匀性

通用代码实现 #

from collections import defaultdict

def distribute_lists(input_lists: list[list]) -> tuple[list[list], list[list]]:
    list_count = len(input_lists)
    # 统计每个元素出现的所有列表索引
    element_occur = defaultdict(list)
    for idx, lst in enumerate(input_lists):
        for ele in lst:
            element_occur[ele].append(idx)
    
    # 筛选重叠元素：出现在≥2个列表中的元素
    overlap_elements = [ele for ele, idxs in element_occur.items() if len(idxs) >=2]
    
    # 初始化Final和Skips列表
    final_lists = []
    skips_lists = []
    for idx, lst in enumerate(input_lists):
        # 初始值是当前列表独有的非重叠元素
        unique_ele = [ele for ele in lst if ele not in overlap_elements]
        final_lists.append(unique_ele)
        skips_lists.append([])
    
    # 均匀分配重叠元素
    for assign_idx, ele in enumerate(overlap_elements):
        # 计算当前元素要分配到的目标列表索引
        target_list_idx = assign_idx % list_count
        # 添加到目标列表的Final
        final_lists[target_list_idx].append(ele)
        # 添加到其他所有列表的Skips
        for i in range(list_count):
            if i != target_list_idx:
                skips_lists[i].append(ele)
    
    return final_lists, skips_lists

# 测试用例：3个列表场景
if __name__ == "__main__":
    Alpha = ['1', '2', '3', '4', '6', '8']
    Bravo = ['1', '2', '3', '5', '7', '9']
    Charlie = ['1', '2', '5', '6', '10', '11']
    
    finals, skips = distribute_lists([Alpha, Bravo, Charlie])
    # 输出结果
    for i, (final, skip) in enumerate(zip(finals, skips)):
        print(f"列表{i+1}Final: {final}")
        print(f"列表{i+1}Skips: {skip}\n")

效果说明 #

• 分配均匀性：重叠元素按索引取模分配，N个列表的重叠元素分配数量差最多为1
• 完全互斥：每个重叠元素仅分配给1个列表，所有Final列表无重复元素
• 适配任意数量列表：支持2个及以上任意数量的输入列表，无需修改逻辑

注意事项 #

• 如果需要严格保留原始列表的元素顺序，可在分配完成后，按照元素在原始输入列表中的出现位置对Final列表进行排序
• 若重叠元素有优先级分配要求，可调整overlap_elements的排序规则，再进行分配

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