5-p05_prisoners

import numpy as np

def prison(prisoners):
    monitor=prisoners - 1 #定义一个班长
    switch=[False] * monitor#出去人数
    lamp = False#开关
    num=0

    while True:
        luck=np.random.randint(0,prisoners)#随机整数
        print(luck)
        if luck== monitor:
            print('---------------')
            if lamp:#灯开着
                lamp=False
                num += 1
                if num == monitor:
                    break
        else:
            if not lamp and not switch[luck]:
                lamp=True
                switch[luck]=True

    print('All prisoners are luck')

if __name__=='__main__':
     prison(4)

D:\Anaconda\python.exe D:/AI20/06_codes/deeplearning_20/p05_prisoners.py
2
1
0
3
---------------
1
2
0
3
---------------
0
2
0
1
0
2
1
0
1
3
---------------
All prisoners are luck

Process finished with exit code 0