Spaces:
Running
Running
File size: 589 Bytes
a4da721 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
with open("input.txt") as f:
data = f.read()
# with open("test.txt") as f:
# data = f.read()
l1 = []
l2 = []
for line in data.split("\n"):
if len(line.split(" ")) > 1:
v1, v2 = line.split(" ")
l1.append(int(v1))
l2.append(int(v2))
total = 0
for n1, n2 in zip(sorted(l1), sorted(l2)):
total += abs(n2 - n1)
print(total)
# Problem 2
def get_counts(l):
counts = {}
for num in l:
counts[num] = counts.get(num, 0) + 1
return counts
c2 = get_counts(l2)
total = 0
for n in l1:
total += n * c2.get(n, 0)
print(total)
|