比赛模拟

from random import random
def printInfo():
print("\t\t这个程序模拟2个队伍A和B的排球竞技比赛！")
print("\t 程序运行需要队伍A和B的能力值(0到1之间的小数表示)")

def getInputs():
a = eval(input("请输入队伍A的能力值(0~1)："))
b = eval(input("请输入队伍B的能力值(0~1)："))
n = eval(input("请输入比赛次数："))
return a, b, n

def NGames(n, probA, probB):
winA, winB = 0, 0
for _ in range(n):
scoreA, scoreB = OneGame(probA, probB)
if scoreA > scoreB:
winA += 1
else:
winB += 1
return winA, winB

def OneGame(probA, probB):
scoreA, scoreB, N = 0, 0, 0
serving = 'A'
while not GameOver(N, scoreA, scoreB):
if serving == 'A':
if random() > probA:
scoreB += 1
serving = 'B'
else:
scoreA += 1
if serving == 'B':
if random() > probB:
scoreA += 1
serving = 'A'
else:
scoreB += 1
N += 1
return scoreA, scoreB

def GameOver(N, a, b):
if N <= 4:
return (a>=25 and b>=25 and abs(a-b)>=2)
else:
return (a>=15 and b>=15 and abs(a-b)>=2)

def printResult(n, winA, winB):
print("竞技分析开始，共模拟{}场比赛".format(n))
print("队伍A获胜{}场比赛，占比{:0.1%}".format(winA,winA/n))
print("队伍B获胜{}场比赛，占比{:0.1%}".format(winB,winB/n))

if __name__ == "__main__":
printInfo()
probA, probB, n = getInputs()
winA, winB = NGames(n, probA, probB)
printResult(n, winA, winB)