From 49329c308630de66cd911edfe759ee82e8f084b5 Mon Sep 17 00:00:00 2001 From: Thomas Voss Date: Fri, 3 Dec 2021 07:13:56 +0100 Subject: Perform epic gamer code compression --- 2021/03/puzzle-2.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to '2021') 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"] -- cgit v1.2.3