AdventOfCode/2024/01/solution.py

35 lines
715 B
Python

list1 = []
list2 = []
file = open("input.txt", "r").read()
lines = file.split("\n")
for line in lines:
nums = line.split(" ")
num1 = int(nums[0])
num2 = int(nums[1])
list1.append(num1)
list2.append(num2)
list1.sort()
list2.sort()
# for i in range(len(list1)):
# print(list1[i], list2[i])
total_dist = 0
for i in range(len(list1)):
dist = list2[i] - list1[i]
dist = abs(dist)
total_dist += dist
print("Total distance: ", total_dist)
total_similarity = 0
for i in range(len(list1)):
num = list1[i]
occurs = 0
for j in range(len(list2)):
if num == list2[j]:
occurs += 1
similarity = num * occurs
total_similarity += similarity
print("Total similarity: ", total_similarity)