Python练习题 017:三支乒乓球队出赛名单

【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)

 

++++++++++++++++++++++++++++++++++++++

题目出处:编程语言入门经典100例【Python版】

posted @ 2016-10-16 21:31  木木卡卡西  阅读(3751)  评论(10编辑  收藏  举报