Python/maths/sock_merchant.py