24 lines
640 B
Python
24 lines
640 B
Python
# SPDX-License-Identifier: MIT
|
|
# Copyright (c) 2019 Akumatic
|
|
#
|
|
# https://adventofcode.com/2019/day/1
|
|
|
|
def readFile() -> list:
|
|
with open(f"{__file__.rstrip('code.py')}input.txt", "r") as f:
|
|
return [int(line[:-1]) for line in f.readlines()]
|
|
|
|
def part1(vals: list) -> int:
|
|
return sum([val // 3 - 2 for val in vals])
|
|
|
|
def part2(vals: list) -> int:
|
|
fuel = [val // 3 - 2 for val in vals]
|
|
for f in fuel:
|
|
temp = f // 3 - 2
|
|
if temp > 0:
|
|
fuel.append(temp)
|
|
return sum(fuel)
|
|
|
|
if __name__ == "__main__":
|
|
vals = readFile()
|
|
print(f"Part 1: {part1(vals)}")
|
|
print(f"Part 2: {part2(vals)}") |