32 lines
436 B
Python
32 lines
436 B
Python
from lib import get_data
|
|
|
|
|
|
def f(i):
|
|
i //= 3
|
|
i -= 2
|
|
return i
|
|
|
|
|
|
def part_1(data):
|
|
print(sum([f(int(line)) for line in data.splitlines()]))
|
|
|
|
|
|
def part_2(data):
|
|
r = 0
|
|
for line in data.splitlines():
|
|
i = f(int(line))
|
|
while i > 0:
|
|
r += i
|
|
i = f(i)
|
|
print(r)
|
|
|
|
|
|
def main():
|
|
data = get_data(__file__)
|
|
part_1(data)
|
|
part_2(data)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|