Python编程题14--随机分配礼物

题目

已知有5个人，每人手上都有一个不同的礼物，现在需要对5个人的礼物重新进行分配（也可以理解为5个人互换礼物），要求重新分配后，每个人不能再拿到自己原来的礼物。

实现思路1

• 利用 字典 来存储最开始的5个人及其礼物，再设置一个新字典，用于存储分配后的结果
• 利用字典的 keys() 和 values() 方法，分别得到 人员列表user_list 和 礼物列表gift_list
• 重新分配时，利用 random 内置模块的 choice() 方法，可以从 gift_list 中随机取出礼物
• 每次均从 user_list 中取第一个人，在字典中查找该人的礼物，比较随机取出的礼物是否是原来的礼物，如果不是，那么就将随机取的礼物分配给该人，否则就将随机取的礼物分配给 user_list 的第二个人
• 将已分配好的人员和礼物，存储到新字典中，并从 user_list 和 gift_list 中，分别删除掉已分配好的人员和礼物
• 执行循环，继续进行分配，直到只剩2个人和2个礼物未被重新分配
• 当 user_list 只剩2个人时，不再从 gift_list 随机取出礼物，而是通过判断 user_list[0] 对应的礼物 gift_list[0] 或 user_list[1] 对应的礼物 gift_list[1] ，是否是自己原来的礼物，如果是则二者进行礼物交换，否则直接进行指定分配

代码实现

import random

def distribute_gift(data):
    result = {}
    user_list = list(data.keys())
    gift_list = list(data.values())
    while True:
        if len(user_list) != 2:
            random_gift = random.choice(gift_list)
            if data[user_list[0]] != random_gift:
                result[user_list[0]] = random_gift
                user_list.remove(user_list[0])
            else:
                result[user_list[1]] = random_gift
                user_list.remove(user_list[1])
            gift_list.remove(random_gift)
        else:
            if (data[user_list[0]] == gift_list[0]) or (data[user_list[1]] == gift_list[1]):
                result[user_list[0]] = gift_list[1]
                result[user_list[1]] = gift_list[0]
            else:
                result[user_list[0]] = gift_list[0]
                result[user_list[1]] = gift_list[1]
            break
    return result

data = {"A": "礼物A", "B": "礼物B", "C": "礼物C", "D": "礼物D", "E": "礼物E"}
print("随机分配后：{}".format(distribute_gift(data)))

实现思路2

• 利用 字典 来存储最开始的5个人及其礼物，再设置一个新字典，用于存储分配后的结果
• 利用字典的 values() 方法，得到 所有礼物列表 list1 ，重新分配后的礼物列表 list2
• 设置一个额外的待分配礼物列表 new_list ，其包括那些在 list1 中但不在 list2 中的礼物
• 对字典进行遍历，遍历时的键key，存储的就是已知的5个人员
• 遍历过程中，再通过 while 循环来分配礼物，每次从 待分配礼物列表 new_list 中随机取出一个礼物，并把该礼物分配给当前要分配的人，如果该礼物恰是这个人原来的礼物，那么继续随机抽取。
• 将已分配好的人员和礼物，存储到新字典中
• 如果分到最后，最后这个礼物恰好是最后一个人的，那么就从已分配好的人中，随机抽一个人来和最后一个人交换礼物，这样就可以保证每个人拿到的不再是自己原来的礼物。

代码实现

import random

def distribute_gift(data):
    result = {}
    list1 = list(data.values())
    for user in data:
        list2 = list(result.values())
        new_list = [i for i in list1 if i not in list2]
        if (len(new_list)) == 1 and (data[user] == new_list[0]):
            last_gift = new_list[0]
            random_user = random.choice(list(result.keys()))
            result[user] = result[random_user]
            result[random_user] = last_gift
            break
        while not result.get(user): # 如果result中没有这个人员，则可分配礼物
            random_gift = random.choice(new_list)
            if data[user] != random_gift:
                result[user] = random_gift
    return result

data = {"A": "礼物A", "B": "礼物B", "C": "礼物C", "D": "礼物D", "E": "礼物E"}
print("随机分配后：{}".format(distribute_gift(data)))