21 lines
367 B
Python
21 lines
367 B
Python
import numpy as np
|
|
|
|
l = np.loadtxt("day1.txt")
|
|
# l = np.array([[3,4],
|
|
# [4,3],
|
|
# [2,5],
|
|
# [1,3],
|
|
# [3,9],
|
|
# [3,3]])
|
|
|
|
a,b = np.sort(l[:,0]),np.sort(l[:,1])
|
|
dist = np.abs(a - b)
|
|
tot = np.sum(dist)
|
|
# print(tot)
|
|
|
|
sim = 0
|
|
for i in a :
|
|
for j in b :
|
|
if i == j:
|
|
sim += i
|
|
print(sim) |