Remove unused import and minor refactor

1 files changed, 6 insertions, 15 deletions

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__":