Python如何实现生成唯一排列并写入文本文件

现有代码存在3个核心问题，无法满足需求：

• 原permute函数仅在控制台打印排列结果，没有返回值，变量r实际为None，直接调用f.write(r)会触发类型报错
• 原逻辑无差别生成全量排列，没有做重复排列剪枝，若输入集合存在重复元素会生成大量重复结果
• 原逻辑试图把所有结果一次性写入文件，面对百万级以上的排列规模会直接占满内存，无法适配大数据量场景

额外提示：36个无重复元素的全排列规模为36! ≈ 3.7×10^41，这个量级不可能通过全量生成后筛选的方式处理，必须在递归生成阶段就做剪枝跳过无效分支，同时采用流式写入的方式写文件，避免全量缓存结果。

核心思路是：递归生成排列时同层跳过重复元素做剪枝，从根源避免生成重复排列；打开文件后每生成一个符合要求的排列就逐行写入，全程不缓存全量结果，内存占用恒定。

def permute_unique_and_write(lst, file_handler, f=0):
    if f >= len(lst):
        # 生成合法排列后直接写入文件，不存入内存列表
        file_handler.write(f"{lst}\n")
        return
    # 记录当前层已经使用过的元素，避免交换重复值生成重复排列
    used = set()
    for s in range(f, len(lst)):
        if lst[s] in used:
            # 同层相同元素已经交换过，直接剪枝跳过
            continue
        used.add(lst[s])
        lst[f], lst[s] = lst[s], lst[f]
        permute_unique_and_write(lst, file_handler, f+1)
        lst[f], lst[s] = lst[s], lst[f]

if __name__ == "__main__":
    input_list = ['Cat', 'Dog', 'Bird', 'Rabbit']
    # 用上下文管理器打开文件，自动处理关闭逻辑，避免句柄泄漏
    with open("Permutation_File.txt", "w", encoding="utf-8") as f:
        permute_unique_and_write(input_list, f)

如果需要像示例中那样只保留特定规则的排列（示例仅保留4条结果），不需要修改整体流式框架，只需要在写入前增加判断逻辑即可：

def permute_unique_and_write(lst, file_handler, f=0):
    if f >= len(lst):
        # 在这里增加自定义筛选条件，不符合的直接跳过不写入
        # 示例条件仅为演示，替换成实际业务的判断规则即可
        if <你的自定义判断逻辑>:
            file_handler.write(f"{lst}\n")
        return
    used = set()
    for s in range(f, len(lst)):
        if lst[s] in used:
            continue
        used.add(lst[s])
        lst[f], lst[s] = lst[s], lst[f]
        # 也可以在递归前加分支剪枝逻辑，提前判断当前分支不可能生成符合要求的结果就直接跳过，进一步提升性能
        permute_unique_and_write(lst, file_handler, f+1)
        lst[f], lst[s] = lst[s], lst[f]

• 剪枝逻辑放在越靠前的递归层，性能提升越明显，对于36个元素的场景，必须通过业务规则剪掉绝大多数无效分支，否则计算量完全不可行
• 全程采用逐行写盘的逻辑，不要把所有排列结果存在列表里最后一次性写入，否则哪怕百万级结果也会占用大量内存，亿级以上直接会触发内存溢出
• 写入文件时指定编码格式，避免不同系统下打开文件出现乱码

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