aboutsummaryrefslogtreecommitdiff
path: root/challenge-104/paulo-custodio/python/ch-2.py
blob: 6a71800c90ddd106f51df9fa668fdefe787b5f39 (plain)
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#!/usr/bin/env python3

# Challenge 104
#
# TASK #2 > NIM Game
# Submitted by: Mohammad S Anwar
# Write a script to simulate the NIM Game.
#
# It is played between 2 players. For the purpose of this task, let assume you
# play against the machine.
#
# There are 3 simple rules to follow:
#
# a) You have 12 tokens
# b) Each player can pick 1, 2 or 3 tokens at a time
# c) The player who picks the last token wins the game

# A plays to win, B plays randomly, A wins 91% of the time
# If both play to win, the second player to play always wins

import sys
import random

matches = 100000
random_play = True
wins = {'A':0, 'B':0}

def draw_to_win(t):
    if t <= 3:
        return t        # win the game
    elif (t % 4)==0:
        return 1        # loose the game
    elif (t % 5)==0:
        return 1        # win the game
    elif (t % 6)==0:
        return 2        # win the game
    elif (t % 7)==0:
        return 3        # win the game
    elif (t % 9)==0:
        return 2        # win the game
    elif (t % 11)==0:
        return 2        # win the game
    else:
        return None     # not reached

def draw_random(t):
    return random.randint(1, 3)

def draw(t):
    global random_play

    if random_play:
        return draw_random(t)
    else:
        return draw_to_win(t)

def play_match():
    global wins
    t = 12
    while t > 0:
        for player in ['A', 'B']:
            if player=='A':
                move = draw_to_win(t)
            else:
                move = draw(t)
            t -= move
            if t <= 0:
                wins[player] += 1
                break

def play_matches():
    for i in range(0, matches):
        play_match()

if sys.argv[1]=="random":
    random_play = True
else:
    random_play = False
play_matches()
a_wins = int(100*wins['A']/(wins['A']+wins['B']))
print(f"A wins {a_wins}% of the matches.")