You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
807 B
33 lines
807 B
import itertools
|
|
import sys
|
|
|
|
|
|
def main():
|
|
series = [int(line.rstrip()) for line in sys.stdin]
|
|
|
|
# Part 1
|
|
frame_len = 25
|
|
for i in range(len(series) - frame_len):
|
|
frame = series[i : i + frame_len]
|
|
n = series[i + frame_len]
|
|
for a, b in itertools.combinations(frame, 2):
|
|
if a + b == n:
|
|
break
|
|
else:
|
|
print("Invalid N:", n)
|
|
break
|
|
|
|
# Part 2
|
|
for i in range(len(series)): # what is indexerror when you know it breaks
|
|
for j in range(1, len(series)):
|
|
frame = series[i : i + j + 1]
|
|
if (s := sum(frame)) == n:
|
|
print("Answer:", min(frame) + max(frame))
|
|
return
|
|
elif s > n:
|
|
break
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|