Formatting

This commit is contained in:
akp 2024-12-03 09:36:08 +00:00
parent f308601bfc
commit ee3a3f9233
No known key found for this signature in database
GPG key ID: CF8D58F3DEB20755

View file

@ -4,39 +4,39 @@ from collections import defaultdict
def parse(instr: str) -> tuple[list[int], list[int]]:
a, b = [], []
for line in instr.splitlines():
ai, bi = line.split(" ")
a.append(int(ai))
b.append(int(bi))
return a, b
def one(instr: str) -> int:
a, b = parse(instr)
a = sorted(a)
b = sorted(b)
acc = 0
for (x, y) in zip(a, b):
acc += abs(y - x)
return acc
def two(instr: str):
a, b = parse(instr)
counts = defaultdict(lambda: 0)
for val in b:
counts[val] = counts[val] + 1
acc = 0
for val in a:
acc += counts[val] * val
return acc
@ -53,4 +53,4 @@ if __name__ == "__main__":
if sys.argv[1] == "1":
print(one(inp))
else:
print(two(inp))
print(two(inp))