import numpy as np
from pathlib import Path

filepath = Path("./data/1_example")
filepath = Path("./data/1_input")

with open(filepath, 'r') as filein:
    data = np.transpose([[int(i) for i in l.split()] for l in filein.readlines()])
data.sort()
data_diff = np.abs(np.diff(data,axis=0).flatten())
data_sum = data_diff.sum()
print(data_sum)


def score_similarity(a: np.ndarray, b: np.ndarray):
    left = a# p.unique(a)
    right = np.array([np.sum(np.where(b==i, 1, 0)) for i in left])
    return np.sum(left*right)
    

print(score_similarity(*data))