蓝桥杯——随意组合

题目

小明被绑架到 X 星球的巫师W那里。
其时，巫师正在玩弄两组数据(2358)和(1467)。他命令小明从组数据中分别取数与另一组中的数配对，共配成4对(组中的每个数必被用到)
小明的配法是:(8,7),(5,6),(3,4),(2,1)
巫师凝视片刻，突然说这个配法太棒了!
因为每个配对中的数字组成两位数，求平方和，无论正倒，居然相等:
87**2+56**2+34**2+21**2=12302
78**2+65**2+43**2+12**2=12302
小明想了想说:“这有什么奇怪呢，我们地球人都知道，随便配配也可以啊!”(8,6),(5,4),(3,1),(2,7)
86**2+54**2+31**2+27**2=12002
68**2+45**2+ 13**2+ 72**2= 12002
巫师顿时凌乱了。。。
请你计算一下，包括上边给出的两种配法，巫师的两组数据一共有多少种配对方案具有该特征。
配对方案计数时，不考虑配对的出现次序。就是说:(8,7),(5,6),(3,4),(2,1)与(5,6),(8,7),(3,4),(2,1)是同一种方案。

分析

本题考的是排列组合，python的permutations库可以很好的应对此类问题，首先我们开两个列表存储两组数据，然后使用循环遍历这两个列表的所有排列组合，判断条件设为a**2+b**2+c**2+d**2==int(str(a)[::-1])**2+int(str(b)[::-1])**2+int(str(c)[::-1])**2+int(str(d)[::-1])**2，这样就很好的把每一个数进行翻转并求出平方和。最后把每一个符合的情况累加。注意：最后我们还要考虑重复的情况，这四个数一共有这四个数的全排列种情况，即4*3*2*1种情况，我们得到的结果直接除以4*3*2*1即可。

代码

from itertools import permutations
num=[2,3,5,8]
num2=[1,4,6,7]
count=0
for i in permutations(num):
  for j in permutations(num2):
    a=i[0]*10+j[0]
    b=i[1]*10+j[1]
    c=i[2]*10+j[2]
    d=i[3]*10+j[3]
    if a**2+b**2+c**2+d**2==int(str(a)[::-1])**2+int(str(b)[::-1])**2+int(str(c)[::-1])**2+int(str(d)[::-1])**2:
      count+=1
print(count//(4*3*2*1))