44 lines
940 B
Python
44 lines
940 B
Python
import re
|
|
import sys
|
|
|
|
PACKET_RE = r"(\d+)x(\d+)x(\d+)\n"
|
|
|
|
def load_packets(filename):
|
|
packets = []
|
|
with open(filename, "r") as f:
|
|
for line in f:
|
|
match = re.fullmatch(PACKET_RE, line)
|
|
a, b, c = match.groups()
|
|
a, b, c = int(a), int(b), int(c)
|
|
packets.append((a, b, c))
|
|
return packets
|
|
|
|
# PART 1
|
|
|
|
def necessary_area(packet):
|
|
a, b, c = sorted(packet)
|
|
return 3*a*b + 2*a*c + 2*b*c
|
|
|
|
def total_wrapping_paper(packets):
|
|
return sum(map(necessary_area, packets))
|
|
|
|
# PART 2
|
|
|
|
def ribbon_length(packet):
|
|
a, b, c = sorted(packet)
|
|
return 2*a + 2*b + a*b*c
|
|
|
|
def total_ribbon_length(packets):
|
|
return sum(map(ribbon_length, packets))
|
|
|
|
def main(filename):
|
|
packets = load_packets(filename)
|
|
print(f"Solutions for {filename}")
|
|
total = total_wrapping_paper(packets)
|
|
print(f"Part 1: {total}")
|
|
total_2 = total_ribbon_length(packets)
|
|
print(f"Part 2: {total_2}")
|
|
|
|
if __name__ == "__main__":
|
|
for filename in sys.argv[1:]:
|
|
main(filename)
|