通过python实现随机交换礼物程序详解

Python  /  管理员 发布于 8年前   239

看到了一个面试题，想了两种解法，不知道符不符合要求，记录如下：

题目：有N个人，每人备一个圣诞礼物，现需要写一个程序，随机交互礼物，要求：自己不能换到自己的礼物，用python实现。

方法一：

构造二维列表存储参与者的名字和所带礼物，使用random.choice()随机选择礼物。

import random lsGiftIn = [['Jack','apple'],['June','ball'],['Mary','card'],['Duke','doll'],['James','egg'],['Tina','flute'],['Tom','coffee']]#存储参与者的姓名和自己带来的礼物lsGiftOut = []#存储交换后的结果n = len(lsGiftIn)#参与人数gifts = [i[1] for i in lsGiftIn]#未分配出去的礼物for x in range(n):  flag = 0  person = lsGiftIn[x][0]  myGift = lsGiftIn[x][1]  if myGift in gifts:    flag = 1    gifts.remove(myGift)  getGift = random.choice(gifts)#随机分配礼物  lsGiftOut.append([person,getGift])  gifts.remove(getGift)  if flag:    gifts.append(myGift) print(lsGiftOut)

方法二：

构造字典存储参与者的姓名和礼物，其中姓名为key，礼物为value，使用字典的popitem()方法随机返回礼物。

dictGiftIn = {'Jack':'apple','Peter':'beer','Tom':'card','Duke':'doll','Mary':'pineapple','James':'flute','Tina':'coffee'}dictGiftOut = {}persons = list(dictGiftIn.keys())for p in persons:  flag = 0#标记自己带来的礼物是否还未分配出去  if p in dictGiftIn:    flag = 1    myGift = dictGiftIn.pop(p)#如果自己带来的礼物还未分配，则去掉该礼物  getGift = dictGiftIn.popitem()#随机返回并移除一对key-value值  dictGiftOut[p] = getGift[1]#得到的礼物  if flag:    dictGiftIn[p] = myGift#将自己的礼物添到未分配礼物中 print(dictGiftOut)#输出礼物分配情况

The End ~

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。