Python------算法题目：模拟蒙蒂霍尔悖论游戏

问题描述：假设你正参加一个有奖游戏节目，并且有3道门可选：其中一个后面是汽车，另外两个后面是山羊。你选择一道门，比如说选择1号门，主持人当然知道每道门后面是什么，并且打开了另一道门，比如说3号门，后面是一只山羊。然后主持人问你“你想改选2号门吗?”那么问题来了，改选的话对你会有利吗?这就是所谓的蒙蒂霍尔悖论游戏，也是一个经典的概率问题。

 

基本思路：使用字典来模拟3道门，使用键表示门的编号，值表示门后的物品。

from random import randrange

def init():
    '''返回一个字典，键为3个门号，值为门后面的物品'''
    result = {i: 'goat' for i in range(3)}
    r = randrange(3)
    #在某个随机的门后面放一辆汽车，其他两个门后面仍然是山羊
    result[r] = 'car'
    return result
def startGame():
    #获取本次游戏中每个门的情况
    doors = init()
    #获取玩家选择的门号
    while True:
        try:
            firstDoorNum = int(input('Choose a door to open:'))
            assert 0<= firstDoorNum <=2
            break
        except:
            print('Door number must be between {} and {}'.format(0, 2))

    #主持人查看另外两个门后的物品情况
    #字典的keys()方法返回结果可以当作集合使用，支持使用减法计算差集
    for door in doors.keys()-{firstDoorNum}:
        #打开其中一个后面为山羊的门
        if doors[door] == 'goat':
            print('"goat" behind the door', door)
            #获取第三个门号，让玩家纠结
            thirdDoor = (doors.keys()-{door, firstDoorNum}).pop()
            change = input('Switch to {}?(y/n)'.format(thirdDoor))
            finalDoorNum = thirdDoor if change=='y' else firstDoorNum
            if doors[finalDoorNum] == 'goat':
                return 'I Win!'
            else:
                return 'You Win.'
while True:
    print('='*30)
    print(startGame())
    r = input('Do you want to try once more?(y/n)')
    if r == 'n':
        break

测试结果：