34 lines
851 B
Python
34 lines
851 B
Python
|
|
#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) |