The hat-check problem solution calculator

hat_check_problem.py

import math


n = int(input())


def some_sum():
    s = 0
    for m in range(1, n+1):
        s += ((-1)**(m-1)) * (1/math.factorial(m))
    return s


P = 1 - some_sum()

print(P)