From 02cf6a051d9629479df7d5a6d308153466563de6 Mon Sep 17 00:00:00 2001 From: Thomas Voss Date: Wed, 8 Dec 2021 01:06:09 +0100 Subject: Remove unused import and minor refactor --- 2021/07/puzzles.py | 21 ++++++--------------- 1 file changed, 6 insertions(+), 15 deletions(-) (limited to '2021/07/puzzles.py') diff --git a/2021/07/puzzles.py b/2021/07/puzzles.py index bfd9bb3..a85016a 100644 --- a/2021/07/puzzles.py +++ b/2021/07/puzzles.py @@ -1,26 +1,17 @@ #!/usr/bin/env python3 -import math +def main() -> None: + with open("input", "r", encoding="utf-8") as f: + data = list(map(int, f.read().split(","))) -def gauss_sum(n: int) -> int: # START PART 1 - return n + gauss_sum = lambda n: n # END PART 1 START PART 2 - return n * (n + 1) // 2 + gauss_sum = lambda n: n * (n + 1) // 2 # END PART 2 - -def main() -> None: - with open("input", "r", encoding="utf-8") as f: - data = list(map(int, f.read().split(","))) - - print( - min( - sum(gauss_sum(abs(i - data[j])) for j in range(len(data))) - for i in range(min(data), max(data) + 1) - ) - ) + print(min(sum(gauss_sum(abs(i - n)) for n in data) for i in range(min(data), max(data) + 1))) if __name__ == "__main__": -- cgit v1.2.3