From 0a5c1f77d01fc0bd166494787f24562e2fd3a9e9 Mon Sep 17 00:00:00 2001 From: Botond Hende Date: Wed, 11 Dec 2024 10:18:23 +0100 Subject: renamed day folders to two digit format --- 2024/day01/solve2.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 2024/day01/solve2.py (limited to '2024/day01/solve2.py') diff --git a/2024/day01/solve2.py b/2024/day01/solve2.py new file mode 100644 index 0000000..8b00942 --- /dev/null +++ b/2024/day01/solve2.py @@ -0,0 +1,20 @@ +sum = 0 + +list1 = [] +list2 = [] + +with open("input") as f: + for line in f.readlines(): + nums = line.split() + list1.append(int(nums[0])) + list2.append(int(nums[1])) + +memoize_count = {} + +for num in list1: + if num not in memoize_count.keys(): + memoize_count[num] = list2.count(num) + + sum += memoize_count[num] * num + +print(sum) \ No newline at end of file -- cgit v1.2.3-70-g09d2