使用itertools生成指定长度的扩展数字排列

from itertools import product

# 生成两位0-9的数字组合
# 例如：(0, 0), (0, 1), ..., (9, 9)
for x, y in product(range(10), repeat=2):
    print(f"{x}{y}") # 打印两位数字，如 "00", "01"

from itertools import product, permutations

def get_expanded_permutations(entry: str) -> list[str]:
    """
    将4位数字字符串扩展为6位，并生成所有可能的排列。
    """
    all_perms = set() # 使用集合存储，自动去重
    for x, y in product(range(10), repeat=2):
        # 将原始4位字符串与两位填充数字组合成一个6位字符串
        new_entry = f"{entry}{x}{y}"

        # 对新的6位字符串进行全排列
        for perm_tuple in permutations(new_entry):
            all_perms.add("".join(perm_tuple)) # 将元组转换为字符串并添加到集合
    return list(all_perms)

# 示例用法
input_code = "1234"
results = get_expanded_permutations(input_code)
print(f"为 '{input_code}' 生成的前10个6位排列（已去重）:")
for i, perm in enumerate(results[:10]):
    print(f"{i+1}: {perm}")
print(f"总共生成了 {len(results)} 个不同的排列。")

import os
import datetime
from itertools import product, permutations

def get_expanded_permutations(entry: str) -> list[str]:
    """
    将4位数字字符串扩展为6位，并生成所有可能的排列。
    返回一个去重后的排列列表。
    """
    all_perms = set()
    for x, y in product(range(10), repeat=2):
        new_entry = f"{entry}{x}{y}"
        for perm_tuple in permutations(new_entry):
            all_perms.add("".join(perm_tuple))
    return list(all_perms)

def process_files(input_filepath: str, output_filepath: str, log_filepath: str):
    """
    从输入文件读取4位代码，生成6位排列，并写入输出文件和日志文件。
    """
    if not os.path.exists(input_filepath):
        print(f"错误: 输入文件 '{input_filepath}' 不存在。")
        return

    print(f"开始处理文件：'{input_filepath}'")
    print(f"结果将写入：'{output_filepath}'")
    print(f"日志将写入：'{log_filepath}'")

    with open(input_filepath, 'r') as infile, \
         open(output_filepath, 'w') as outfile, \
         open(log_filepath, 'w') as logfile:

        logfile.write(f"排列生成日志 - {datetime.datetime.now()}\n\n")

        input_data = [line.strip() for line in infile if line.strip()] # 读取并清理输入数据

        total_entries = len(input_data)
        processed_count = 0

        for entry in input_data:
            if not entry.isdigit() or len(entry) != 4:
                logfile.write(f"警告: 跳过无效输入 '{entry}' (非4位数字)。\n")
                continue

            perms = get_expanded_permutations(entry)

            # 将所有排列用换行符连接，一次性写入文件
            if perms:
                outfile.write("\n".join(perms))
                outfile.write("\n") # 确保下一个条目的排列从新行开始

            logfile.write(f"为条目 '{entry}' 生成了 {len(perms)} 个排列。\n")

            processed_count += 1
            print(f"已处理 {processed_count}/{total_entries} 个条目。")

    print("排列生成成功！")

# 假设输入文件名为 input.txt，包含如下内容：
# 1234
# 5678
# 9012

if __name__ == "__main__":
    # 创建一个虚拟的 input.txt 文件用于测试
    with open("input.txt", "w") as f:
        f.write("1234\n")
        f.write("5678\n")
        f.write("9012\n")
        f.write("ABCD\n") # 无效输入示例
        f.write("123\n")  # 无效输入示例

    current_dir = os.path.dirname(os.path.abspath(__file__))
    input_file_path = os.path.join(current_dir, "input.txt")
    output_file_path = os.path.join(current_dir, "output.txt")
    log_file_path = os.path.join(current_dir, f"permutation_log_{datetime.datetime.now().strftime('%Y%m%d%H%M%S')}.log")

    process_files(input_file_path, output_file_path, log_file_path)