看到了一个面试题,想了两种解法,不知道符不符合要求,记录如下:
题目: 有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 ~
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。