Solve 2015/02

2015/02/solve.py

@@ -0,0 +1,44 @@
+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)