advent-of-code/2020/day1/day1.py

16 lines
394 B
Python
Raw Normal View History

2020-12-01 16:40:10 +01:00
#!/usr/bin/env python
import sys
from typing import Iterable
def find2020(nums: Iterable[int]) -> int:
candidates = set()
for num in nums:
opposite = 2020-num
if num in candidates:
return num * opposite
candidates.add(opposite)
if __name__ == '__main__':
with open(sys.argv[1], 'r') as f:
print(find2020(int(n) for n in f.readlines()))