from random import random
def printIntro():
print("这个程序模拟两个选手A和B的乒乓球竞技比赛 \"10\"")
print("程序运行需要传输A和B的能力值(以0到1之间的小数表示)")
#获取输入值
def getInputs():
a = eval(input("请输入选手A的能力值(0-1): "))
b = eval(input("请输入选手B的能力值(0-1): "))
c = eval(input("请输入为单打还是双打,单打为1,双打为2:"))
return a, b, c
#判断单打双打并返回进行的局数
def sdRule(c):
if c==1:
return 7
else:
return 5
def simNGames(n, c, probA, probB):
winsA, winsB = 0, 0
for i in range(n):
scoreA, scoreB = simOneGame(probA, probB)
if scoreA > scoreB:
winsA += 1
else:
winsB += 1
if allGameOver(c,winsA,winsB):
break
return winsA, winsB
#判断一场比赛是否结束
def allGameOver(c,a,b):
if c ==1:
return a == 4 or b == 4
else:
return a == 3 or b == 3
#判断一局是否结束
def gameOver(a,b):
if (a>=11 and b<11) or (b>=11 and a<11) or (a>9 and b>9 and ((a-b)>=2 or (b-a)>=2)):
return True
return False
#判断一局谁能赢
def simOneGame(probA, probB):
scoreA, scoreB = 0, 0
serving = "A"
print('比赛开始了,')
while not gameOver(scoreA, scoreB):
if serving == "A":
if random() < probA:
scoreA += 1
print('A取得了胜利')
else:
serving="B"
else:
if random() < probB:
scoreB += 1
print('B取得了胜利')
else:
serving="A"
return scoreA, scoreB
def printSummary(winsA, winsB):
n = winsA + winsB
print("竞技分析开始,共模拟{}场比赛".format(n))
print("选手A获胜{}场比赛,占比{:0.1%}".format(winsA, winsA/n))
print("选手B获胜{}场比赛,占比{:0.1%}".format(winsB, winsB/n))
def main():
printIntro()
probA, probB, c = getInputs()
winsA, winsB = simNGames(sdRule(c), c, probA, probB)
printSummary(winsA, winsB)
main()