2017年東大理系数学をpythonで解いてみた！ - 東大理三生のプログラミング奮闘記

昨日、今日で東大入試がありましたね！

受験生の方、受験お疲れ様でした。

せっかくなので、数学の問題をpythonで解いてみよう！ということで解いてみました！

pythonの機能（ライブラリ）がまだ全然使いこなせないので、今回は第２問（確率）だけを解いてみました。

（問題）
f:id:gragragrao:20170226144148p:plain

（解答）

# 2017年　東大数学　理系

# 第２問

import itertools
from fractions import Fraction

move_pattern = ["a", "b", "c", "d"]
move_command = list(itertools.product(move_pattern, repeat=6))

whole_event = 4 ** 6
counter_1 = 0
counter_2 = 0

for command in move_command:
    coordinate = [0, 0]
    for s in command:
        if s == "a":
            coordinate[0] += 1
        if s == "b":
            coordinate[0] -= 1
        if s == "c":
            coordinate[1] += 1
        if s == "d":
            coordinate[1] -= 1
    if coordinate[0] == coordinate[1]:
        counter_1 += 1

prob_1 = Fraction(counter_1, whole_event)
print("the answer of (1) is {}".format(prob_1))


for command in move_command:
    coordinate = [0, 0]
    for s in command:
        if s == "a":
            coordinate[0] += 1
        if s == "b":
            coordinate[0] -= 1
        if s == "c":
            coordinate[1] += 1
        if s == "d":
            coordinate[1] -= 1
    if coordinate[0] == 0 and coordinate[1] == 0:
        counter_2 += 1

prob_2 = Fraction(counter_2, whole_event)
print("the answer of (2) is {}".format(prob_2))