summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--2024/day1/solve2.py20
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