diff --git a/2024/1/list_distances.py b/2024/1/list_distances.py index f429277..3e65642 100644 --- a/2024/1/list_distances.py +++ b/2024/1/list_distances.py @@ -1,26 +1,21 @@ -import fileinput +import fileinput -list1 = []; -list2 = []; -for line in fileinput.input(files='input', encoding="utf-8"): - list = line.rstrip().partition(' ') - list1.append(list[0]) - list2.append(list[2]) +def main(): + left_values = [] + right_values = [] -list1.sort() -list2.sort() + for line in fileinput.input(files="input", encoding="utf-8"): + left, right = line.split() + left_values.append(int(left)) + right_values.append(int(right)) -listSums = []; + left_values.sort() + right_values.sort() -for x in range(len(list1)): - listSums.append(abs(int(list1[x]) - int(list2[x]))) - -distance = 0; - -for num in listSums: - distance += num; - -print(distance) + distance = sum(abs(a - b) for a, b in zip(left_values, right_values)) + print(distance) +if __name__ == "__main__": + main()