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

15 lines
394 B
Python
Executable file

#!/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()))