看到了一个面试题,想了两种解法,不知道符不符合要求,记录如下:
题目:有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