田忌赛马问题可能赢的问题假设,假设田忌和大王的马的等级相同时,大王赢,如果有多个马,会有多少种情况呢。
from itertools import combinations, permutations
import numpy as np
#有几个马
num_house=8
#生成马的列表
items=range(1,num_house+1)
#print(list(permutations(items, len(items))))
#使用permutations(items, len(items) 生成排列组合
tianji_houses = np.array(list(permutations(items, len(items))))
#大王出场次序
wang_houses=np.array(items ).ravel()
#每一场出场次数,然后用相关值的差来进行比较,如果大于0则这场次田忌胜。
#返回的结果为true or false ,进行值的转换为整形
result_first= (tianji_houses-wang_houses>0).astype(int)
#如果所得到值大于场次的一半,则胜利
win_nums=(np.sum(result_first,axis=1)>=(num_house//2+1)).astype(int)
#打出可能胜利的场次
print(np.sum(win_nums))