diff options
-rwxr-xr-x | 2021/03/puzzle-2.py | 8 |
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"] |