48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
import os
|
|
|
|
|
|
def get_package_sides(x: str) -> int:
|
|
l, w, h = map(int, x.split("x")) # map( int, ...) is necessary, to cast the strings to ints
|
|
lw, wh, hl = l*w, w*h, h*l
|
|
smallest_side = min(lw, wh, hl)
|
|
all_sides = (lw + wh + hl) * 2
|
|
return smallest_side + all_sides
|
|
|
|
|
|
def calculate_all_packages(x: list) -> int:
|
|
whole_size = 0
|
|
for box in x:
|
|
whole_size += get_package_sides(box)
|
|
return whole_size
|
|
|
|
|
|
def get_package_ribbon(x: str) -> int:
|
|
l, w, h = map(int, x.split("x"))
|
|
bow = l * w * h
|
|
lw, wh, hl = l*2 + w*2, w*2 + h*2, h*2 + l*2
|
|
smallest_perimeter = min(lw, wh, hl)
|
|
return smallest_perimeter + bow
|
|
|
|
|
|
def calculate_all_packages_ribbon(x: list) -> int:
|
|
whole_ribbon = 0
|
|
for box in x:
|
|
whole_ribbon += get_package_ribbon(box)
|
|
return whole_ribbon
|
|
|
|
|
|
if __name__ == '__main__':
|
|
assert get_package_sides("2x3x4") == 58, "Error: Example 1 couldn't be solved"
|
|
assert get_package_sides("1x1x10") == 43, "Error: Example 2 couldn't be solved"
|
|
|
|
assert get_package_ribbon("2x3x4") == 34, "Error: Part2: Example 1 couldn't be solved"
|
|
assert get_package_ribbon("1x1x10") == 14, "Error: Part2: Example 2 couldn't be solved"
|
|
|
|
print("All tests passed!")
|
|
puzzle_input = open("input.txt", "r").read().splitlines()
|
|
print("input: ", puzzle_input)
|
|
print("solution: ", calculate_all_packages(puzzle_input))
|
|
|
|
print("Part 2:")
|
|
print("solution: ", calculate_all_packages_ribbon(puzzle_input))
|