Python笔记08-乒乓球队赛程安排-回溯法

题目是 两个乒乓球队进行比赛,各出三人。甲队为a,b,c三人,乙队为x,y,z三人。已抽签决定比赛名单。有人向队员打听比赛的名单。a说他不和x比,c说他不和x,z比。请编程序找出三队赛手的名单

这里我用的是回溯法，代码如下：

def get_count(x):
    return len(rul[x])
group1 = ['a','b','c']
group2 = ['x','y','z']
# 这个地方后期可以封装起来用for循环建立，然后在删除元素
rul = {
    'a':['y','z'],
    'b':['x','y','z'],
    'c':['y']
}
# 这里可以用深拷贝来做
choice = {
    'a':['y','z'],
    'b':['x','y','z'],
    'c':['y']
}
# 记录已做选择的队员
seen_members = []
# 记录被选择的对手
seen_opps = []
# 存放结果
res = {}
times = 0
group1.sort(key=get_count)#让选择少的排在前面
while group1:
    # 第一个队员出队列
    member = group1.pop(0)
    # 列出可选对手
    opps = choice[member]
    # 遍历对手
    for opp in opps:
        times +=1
        if opp not in seen_opps:
            # 各种记录
            res[member] = opp#记录选择
            seen_opps.append(opp)#记录被选择对手
            seen_members.append(member)#记录做选择队员
            choice[member].remove(opp)#减少队员的一个选择
            break
    else:#如果该选手没做选择，说明上个选手选错了
        re_member = seen_members.pop()#取出上一个选手
        group1.append(member)#将刚出队的放回去
        group1.insert(0,re_member)#让上一个重选
        choice[member] = rul[member]#选择初始化
        seen_opps.pop()#踢出上一个选手选择的对手
print(res)
print(times)