2024 day1 solution2

1 files changed, 20 insertions, 0 deletions

diff --git a/2024/day1/solve2.py b/2024/day1/solve2.py
new file mode 100644
index 0000000..8b00942
--- /dev/null
+++ b/2024/day1/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