Perform epic gamer code compression

1 files changed, 4 insertions, 4 deletions

diff --git a/2021/03/puzzle-2.py b/2021/03/puzzle-2.py
index c5ae894..55513dd 100755
--- a/2021/03/puzzle-2.py
+++ b/2021/03/puzzle-2.py
@@ -10,10 +10,10 @@ def solve(lines: list[str], comp: Callable[[int, int], bool]) -> int:
 			if len(lines) == 1:
 				return int(lines[0], 2)
 
-			ones = len([line for line in lines if line[i] == "1"])
-			zeros = len([line for line in lines if line[i] == "0"])
-
-			if comp(zeros, ones):
+			if comp(
+				len([line for line in lines if line[i] == "0"]),
+				len([line for line in lines if line[i] == "1"]),
+			):
 				lines = [line for line in lines if line[i] == "0"]
 			else:
 				lines = [line for line in lines if line[i] == "1"]