看到了一个面试题,想了两种解法,不知道符不符合要求,记录如下:
题目:有N个人,每人备一个圣诞礼物,现需要写一个程序,随机交互礼物,要求:自己不能换到自己的礼物,用python实现。
方法一:
构造二维列表存储参与者的名字和所带礼物,使用random.choice()随机选择礼物。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
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()方法随机返回礼物。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
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 ~
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/u013378642/article/details/89684733