Add files via upload

jrmdev1 · web-flow · commit 9e8b1b12b041 · 2021-12-05T14:02:03.000-06:00
diff --git a/3b.py b/3b.py
@@ -0,0 +1,60 @@
+#Advent of code 2021
+# 12/03/21 day 3b
+# Joe McFarland
+# import sys
+# import re
+
+filename = "data3.txt"
+
+file = open(filename)
+filestr = file.read()
+a_list = filestr.split("\n")
+maxrows = len(a_list)
+#print(a_list)
+
+# def read_bit(x, pos):
+#     mask = 1 << pos
+#     return (x & mask) >> pos
+
+def arraytostr(arr):
+    mystr = ""
+    for colval in (arr):
+        if colval == 0:
+            mystr += "0"
+        elif colval == 1:
+            mystr += "1"
+        else:
+            print(f"ERROR {arr}")
+            exit()
+    return mystr
+
+maxcols = len(a_list[0])
+gamma = [None]*maxcols
+epsilon = [None]*maxcols
+for col in range(maxcols):
+    zero_bits = one_bits = 0
+    for row in a_list:
+        if row[col] == "0":
+            zero_bits += 1
+        elif row[col] == "1":
+            one_bits += 1
+        else:
+            print(f"ERROR {row}")
+            exit()
+    if zero_bits > one_bits:
+        gbit = 0
+        ebit = 1
+    else:
+        gbit = 1
+        ebit = 0
+    gamma[col] = gbit
+    epsilon[col] = ebit
+
+print(f"gamma={gamma}, epsilon={epsilon}") 
+g1 = arraytostr(gamma)
+e1 = arraytostr(epsilon)
+gamma_dec = int(g1,2)
+epsilon_dec = int(e1,2)
+print(f"gamma_str={g1}, epsilon_str={e1}") 
+print(f"gamma_dec={gamma_dec}, epsilon_dec={epsilon_dec}") 
+print(f"mul = {gamma_dec*epsilon_dec}")
