From 934b34970332cb6676001d950ca758e005621836 Mon Sep 17 00:00:00 2001 From: kwout Date: Sat, 3 Sep 2022 20:48:58 -0400 Subject: [PATCH] streamlined 3a --- day3/3a.py | 21 +++++++++------------ 1 file changed, 9 insertions(+), 12 deletions(-) diff --git a/day3/3a.py b/day3/3a.py index 64f9cbf..b15c55b 100644 --- a/day3/3a.py +++ b/day3/3a.py @@ -1,25 +1,22 @@ lines = open("input.txt", 'r').read().splitlines() leng = len(lines) -count = [0] * len(lines[0]) +bits = len(lines[0]) +count = [0] * bits for s in lines: - for i in range(0, len(s)): + for i in range(0, bits): if bool(int(s[i])): count[i] += 1 -s = "" -t = "" +gama = 0 +epsi = 0 print(count) -for i in count: - if i > leng/2: - s += "1" - t += "0" +for i in range(0, bits): + if count[i] > leng/2: + gama += 2**(bits-i-1) else: - s += "0" - t += "1" + epsi += 2**(bits-i-1) -gama = int(s, 2) -epsi = int(t, 2) print(gama) print(epsi) print(gama*epsi)