advent-of-code/2015/d20/ex2/ex2.py

26 lines
588 B
Python
Executable file

#!/usr/bin/env python
import sys
def solve(input: str) -> int:
needed = int(input.strip())
presents = [0] * (needed // 10) # Surely this should be enough houses
for i in range(1, len(presents)):
for j in range(1, 50 + 1):
if (i * j) >= len(presents):
continue
presents[i * j] += 11 * i
for i, total in enumerate(presents):
if total >= needed:
return i
assert False # Sanity check
def main() -> None:
input = sys.stdin.read()
print(solve(input))
if __name__ == "__main__":
main()