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__":