From 0fd7b3dc96c8dc8ec93ee6e7860379ea220a951f Mon Sep 17 00:00:00 2001 From: kirbylife Date: Sun, 1 Dec 2024 23:44:12 -0600 Subject: [PATCH] Add day part 2 --- day-1/main.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/day-1/main.py b/day-1/main.py index 32a1121..2f4a3b6 100644 --- a/day-1/main.py +++ b/day-1/main.py @@ -10,7 +10,7 @@ def parse_input() -> tuple[list[int], list[int]]: line2.append(int(num2)) return line1, line2 -def main() -> int: +def part_1(): line1, line2 = parse_input() line1.sort() line2.sort() @@ -21,6 +21,22 @@ def main() -> int: print(result) +def part_2(): + line1, line2 = parse_input() + + appears = {} + for line in line2: + appears[line] = appears.get(line, 0) + 1 + + result = sum(appears.get(num, 0) * num for num in line1) + + print(result) + +def main() -> int: + print("Part 1: ", end="") + part_1() + print("Part 2: ", end="") + part_2() return 0 if __name__ == "__main__":