-
Notifications
You must be signed in to change notification settings - Fork 0
/
left_right_center.py
52 lines (39 loc) · 1.27 KB
/
left_right_center.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
import random
import pprint
from statistics import mean
# from https://fivethirtyeight.com/features/hark-two-holiday-puzzles/
player_count = 6
bills = 3
simulations = 100000
class Player:
def __init__(self):
self.bills = bills
def game():
turns = 0
players = [Player() for _ in range(6)]
current_player = 0
def turn(player_index):
rolls = [random.randint(1, 6) for _ in range(3)]
for roll in rolls:
if players[player_index].bills > 0:
if roll <= 2:
players[player_index - 1].bills += 1
elif roll <= 4:
players[(player_index + 1) % player_count].bills += 1
players[player_index].bills -= 1
while len([player for player in players if player.bills > 0]) > 1:
turn(current_player)
current_player = (current_player + 1) % player_count
turns += 1
return turns
# show results
all_results = [game() for _ in range(simulations)]
print(f"Expected turns per game: {mean(all_results)}")
# shows number of turns, and how many games ended after that number of turns
results = {}
for result in all_results:
if result not in results:
results[result] = 1
else:
results[result] += 1
pprint.pprint(results)