38 lines
1 KiB
Python
Executable file
38 lines
1 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# https://adventofcode.com/2024/day/1
|
|
|
|
def get_lines(filename: str) -> list:
|
|
with open(filename, "r") as file:
|
|
return [line.strip() for line in file.readlines()]
|
|
|
|
|
|
def main():
|
|
# lines = get_lines("sample-input.txt")
|
|
lines = get_lines("input.txt")
|
|
list_a = []
|
|
list_b = []
|
|
for line in lines:
|
|
(a, b) = line.split(" ")
|
|
list_a.append(int(a))
|
|
list_b.append(int(b))
|
|
sorted_a = sorted(list_a)
|
|
sorted_b = sorted(list_b)
|
|
# print(sorted_a)
|
|
# print(sorted_b)
|
|
|
|
total_distance = 0
|
|
for i in range(len(sorted_a)):
|
|
total_distance += abs(sorted_b[i] - sorted_a[i])
|
|
print("Part 1: Total distance is", total_distance)
|
|
|
|
similarity_score = 0
|
|
for i in range(len(list_a)):
|
|
search_val = list_a[i]
|
|
appearances = [j for j, x in enumerate(list_b) if x == search_val]
|
|
similarity_score += search_val * len(appearances)
|
|
print("Part 2: Similarity score is", similarity_score)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|