Python中生成两个列表的多组唯一全配对组合的技术问题

你的需求本质是构造一个n阶拉丁方（这里n=12）：每组是列表A到列表B的一个完美匹配（无重复配对），且所有组叠加后，A中每个元素和B中每个元素恰好配对一次。用itertools.permutations直接生成所有排列再筛选完全不现实——12!是4.79e8级别的数，内存必然溢出，得用直接构造的方法。

核心思路：循环移位构造法 #

这种方法无需生成所有排列，直接通过对列表B的循环移位来生成每组配对，时间复杂度O(n²)，内存占用极低。

示例演示（n=4） #

对应你给出的示例：

• 初始组：A元素按顺序对应B的原顺序
• 后续每组：将B列表循环右移（或左移）固定步数，再和A元素按顺序配对

用代码实现示例的逻辑：

list_a = ['a','b','c','d']
list_b = ['blue','green','red','white']
n = len(list_a)

# 生成n组配对
result = []
for shift in range(n):
    # 循环右移shift位：用切片实现
    shifted_b = list_b[-shift:] + list_b[:-shift]
    # 按顺序配对A和移位后的B
    group = [[a, b] for a, b in zip(list_a, shifted_b)]
    result.append(group)

# 输出结果
print(result)

运行后得到的结果和你给出的示例完全一致（可调整移位方向改变组的顺序）。

适配12元素列表的代码 #

直接替换成你的目标列表即可：

list_a = [f"a_{i}" for i in range(12)]  # 替换为你的列表A元素
list_b = [f"b_{i}" for i in range(12)]  # 替换为你的列表B元素
n = len(list_a)

full_pairs = []
for shift in range(n):
    # 循环右移shift位，生成当前组的B序列
    shifted_b = list_b[-shift:] + list_b[:-shift]
    # 生成当前组的配对，转成列表格式
    current_group = [list(pair) for pair in zip(list_a, shifted_b)]
    full_pairs.append(current_group)

正确性验证 #

可以用以下代码确认所有配对仅出现一次：

from collections import defaultdict

pair_count = defaultdict(int)
for group in full_pairs:
    for a, b in group:
        pair_count[(a, b)] += 1

# 验证所有配对次数为1，且总配对数为n²
assert all(v == 1 for v in pair_count.values())
assert len(pair_count) == n * n

这种方法完全避免了生成海量排列，内存占用仅为存储12组配对的空间，对于n=12的场景毫无压力。

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