#INPUT = './day01/sampleinput.txt'
INPUT = './day01/input.txt'

def parse_input(file_input:str):
    leftcol = []
    rightcol = []
    with open(file_input) as ifile:
        for line in ifile.readlines():
            spair = line.strip().split('   ')
            leftcol.append(int(spair[0]))
            rightcol.append(int(spair[1]))
    return leftcol, rightcol

def p1_solve(left, right):
    left.sort()
    right.sort()
    result = 0
    for i in range(len(left)):
        result += abs(left[i] - right[i])
    print('Answer to part 1 is: {}'.format(result))

def p2_solve(left, right):
    result = 0
    for i in range(len(left)):
        num = left[i]
        result += num * right.count(num)
    print('Answer to part 2 is: {}'.format(result))


if __name__ == '__main__':
    l,r = parse_input(INPUT)
    p1_solve(l,r)
    p2_solve(l,r)