【Python练习题 017】 两个乒乓球队进行比赛,各出三人。甲队为a,b,c三人,乙队为x,y,z三人。已抽签决定比赛名单。有人向队员打听比赛的名单。a说他不和x比,c说他不和x,z比。请编程序找出三队赛手的名单。
------------------------------------------------------
这题真真想破我脑袋了,看了好几份别人的代码才勉强看懂,真是…… 一开始我只想着先把所有可能都配出来(ax, ay, az, bx, by, bz, cx, cy, cz),然后根据后面的条件排除不可能的配对。怎么写也写不对。
后来看了别人的代码,结合自己的思考,整理出基本思路如下:假设a,b,c的对手分别是i,j,k,将i,j,k所有可能出现的组合先穷举出来,同时需要满足2个条件:i,j,k不能同时出现(即a,b,c的对手不可能有重复);a不对x,c不对x,z,只要满足此2条件就可以。简而言之:用3个for穷举,用2个if限定条件。代码如下:
for i in range(ord('x'), ord('z')+1): #假设a,b,c的对手分别是i,j,k
for j in range(ord('x'), ord('z')+1): #用3个for穷举i,j,k可能出现的所有组合
for k in range(ord('x'), ord('z')+1):
if i != j and j != k and k != i: #条件1:i,j,k不能同时出现
if i != ord('x') and k != ord('x') and k != ord('z'): #条件2:a不对x,c不对x,z
print('a vs %s, b vs %s, c vs %s' % (chr(i), chr(j), chr(k)))
输出结果如下:
a vs z, b vs x, c vs y
根据上述代码,联想到之前 求解不重复的3位数 时用到的itertools库,觉得可以简化代码如下:
import itertools for i in itertools.permutations('xyz'):
if i[0] != 'x' and i[2] != 'x' and i[2] != 'z':
print('a vs %s, b vs %s, c vs %s' % (i[0], i[1], i[2]))
输出结果一样。代码简洁很多,结构也清晰不少。
itertools.permutations()是个好东西,可以直接将'xyz'生成包含所有排列的列表。另外两个相关的库方法分别是:itertools.product()生成笛卡尔数列(即包括所有排列方式,itertools.combinations()生成所有组合的列表。就生成项的数目而言,product() > permutations() > combinations()。本题中,permutations('xyz', 3)可生成6组排列方式,而如果换成combinations('xyz', 3),则只有1种组合方式('x', 'y', 'z')。详见官方文档。
【2016-10-17 更新】----------------------------------------------------
感谢 rm-rf 的耐心解答,又多了一种解题思路:先将 a,b,c 这一队列出所有可能的排列方式(共6组),然后每一组都跟 x,y,z 进行匹配(用zip()方法),并设置判断条件。事实证明,这6组之中,只有1组是满足条件的。代码如下:
import itertools team_1 = ['a', 'b', 'c']
team_2 = ['x', 'y', 'z'] for i in itertools.permutations(team_1, 3):
for j in zip(i, team_2):
if j in [('a','x'),('c','x'),('c','z')]:
break
else:
print(i, team_2)
输出结果如下:
('b', 'c', 'a') ['x', 'y', 'z']
rm-rf 还提供了另一种写法,但是……好吧,还没学到 lambda,自定义函数也还没怎么学,所以根本看不懂…… 但还是很感谢啊!希望过不了多久,我就能看懂了~~~
import itertools check_list = [('a', 'x'), ('c', 'x'), ('c', 'z')]
for i in itertools.permutations(team_1, 3):
f = lambda a,b: len([True for j in zip(a, b) if j not in check_list])
if f(i, team_2) == 3:
print(i, team_2)
++++++++++++++++++++++++++++++++++++++