streamlined 3a

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)