Python随机选择室友排班函数遇KeyboardInterrupt且无输出问题求助

原代码的核心问题 #

• 递归逻辑完全错误：函数里的递归调用fair1(a2-1)会让每次调用都重新初始化室友列表，之前的选中记录全丢了，而且递归深度太大还会触发栈溢出，同时直接打断了while循环，根本没法正常执行循环逻辑。
• 计数逻辑彻底跑偏：liste1.count("a")统计的是当前列表里"a"的个数，不是已经选中的次数，完全达不到「统计各元素出现次数」的目的。
• 死循环陷阱：while x1 >= a2/4这类循环里，x1是修改列表前的count结果，不会跟着liste1的变化更新，一旦满足条件就会无限循环，这就是你遇到KeyboardInterrupt的原因——程序卡死，你手动终止了它。
• 没保存选中结果：random.choice(liste1)选完元素后没存下来，最后自然输出不了排班计划。
• 占比判断逻辑错误：应该用「已选中次数 / 总排班次数」判断是否超1/4，不是用剩余次数算。

修复后的代码 #

import random

def generate_clean_schedule(total_times, roommates):
    # 初始化计数字典，记录每个室友被选中的次数
    count = {name: 0 for name in roommates}
    schedule = []
    last_selected = None  # 可选：用来避免连续选中同一人

    for _ in range(total_times):
        # 筛选候选：已选中次数不超总次数1/4，同时避免连续选中（可选）
        candidates = [
            name for name in roommates 
            if count[name] <= total_times / 4 and name != last_selected
        ]
        # 随机选一个符合条件的室友
        selected = random.choice(candidates)
        # 更新计数、排班列表和上一次选中记录
        count[selected] += 1
        schedule.append(selected)
        last_selected = selected
    
    return schedule

# 获取用户输入并生成排班
total = int(input("请输入排班次数："))
roommates_list = ["a", "b", "c", "d"]
result = generate_clean_schedule(total, roommates_list)

# 输出结果
print("清洁排班计划：")
for idx, name in enumerate(result, 1):
    print(f"第{idx}次：{name}")

print("\n各室友排班次数统计：")
for name, num in count.items():
    print(f"{name}：{num}次")

代码说明 #

• 计数机制：用字典count专门记录每个室友的选中次数，替代原代码中错误的列表count方法。
• 候选筛选：每次循环都根据总次数的1/4阈值筛选可选中的室友，保证每人次数相近；额外加了避免连续选中的逻辑，不需要可以删掉对应的判断条件。
• 循环替代递归：用for循环遍历指定次数，避免递归带来的状态丢失和栈溢出问题。
• 结果留存：用schedule列表存储每次选中的结果，最终输出完整排班计划和次数统计。

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