mit Lösung für 14 von E.

14/Day14-Eric.py

@@ -0,0 +1,150 @@
+from numba import jit
+import numpy as np
+from time import time
+
+
+file = open("input.txt").readlines()
+data = []
+for i in file:
+    i = i.strip()
+    bla = []
+    for j in i:
+        bla.append(j)
+    data.append(bla)
+np_data = np.asarray(data)  
+#1_000_000_000
+Part_2 = True
+@jit(nopython = True)
+def roll(arr, n, part):
+    for i in range(n):
+        #roll north
+        for line in range(1,len(arr)):
+            for char in range(len(arr[line])):
+                if arr[line][char] == "O":
+                    p1 = line
+                    p2 = char
+                    while p1-1 >= 0 and arr[p1-1][p2] == ".":
+                        p1-=1
+                    arr[line][char], arr[p1][p2] = arr[p1][p2], arr[line][char]
+        if part:
+            #roll west
+            for line in range(len(arr)):
+                for char in range(1, len(arr[line])):
+                    if arr[line][char] == "O":
+                        p1 = line
+                        p2 = char
+                        while p2-1 >= 0 and arr[p1][p2-1] == ".":
+                            p2-=1
+                        arr[line][char], arr[p1][p2] = arr[p1][p2], arr[line][char]
+            #roll south
+            for line in range(len(arr)-2,-1,-1):
+                for char in range(len(arr[line])):
+                    if arr[line][char] == "O":
+                        p1 = line
+                        p2 = char
+                        while p1+1 < len(arr) and arr[p1+1][p2] == ".":
+                            p1+=1
+                        arr[line][char], arr[p1][p2] = arr[p1][p2], arr[line][char]
+            #roll east      
+            for line in range(len(arr)):
+                for char in range(len(arr[line])-2,-1,-1):
+                    if arr[line][char] == "O":
+                        p1 = line
+                        p2 = char
+                        while p2+1 < len(arr[line]) and arr[p1][p2+1] == ".":
+                            p2+=1
+                        arr[line][char], arr[p1][p2] = arr[p1][p2], arr[line][char]
+
+    result = 0
+    for line in range(len(arr)):
+        line_num = len(arr) - line
+        count_O = 0
+        for i in arr[line]:
+            if i == "O":
+                count_O+=1
+        result+=count_O*line_num
+    return result, arr
+
+
+def finde_laengste_folge(lst):
+    n = len(lst)
+
+    if n == 0:
+        return [], []
+
+    aktuelle_folge = []
+    laengste_folge = []
+    aktuelle_indices = []
+    laengste_indices = []
+
+    for i in range(n):
+        curr_num = lst[i]
+        pointer = i + 1
+        aktuelle_folge.append(curr_num)
+        aktuelle_indices.append(i)
+        repeat = False
+        while pointer+1 < len(lst) and (lst[pointer+1] != curr_num or len(aktuelle_folge) < 4):
+            aktuelle_folge.append(lst[pointer])
+            aktuelle_indices.append(pointer)            
+            pointer+=1
+            if len(aktuelle_folge) > 100:
+                aktuelle_folge = []
+                aktuelle_indices = []
+                repeat = True
+                break
+
+        if repeat:
+            continue
+
+        try:
+            aktuelle_folge.append(lst[pointer])
+            aktuelle_indices.append(pointer)
+        except IndexError:
+            aktuelle_folge.append(lst[-1])
+            aktuelle_indices.append(len(lst)-1)
+        korr_folge = True
+        p1 = 0
+        p2 = i+len(aktuelle_folge)
+        # Prüfe ob sich die gefundene Folge widerholt
+        while p1 < len(aktuelle_folge):
+            try:
+                if aktuelle_folge[p1] != lst[p2]:
+                    korr_folge = False
+                    break
+            except IndexError:
+                korr_folge = False
+                break
+            p1+=1
+            p2+=1
+        
+        if len(aktuelle_folge) > len(laengste_folge) and korr_folge:
+            laengste_folge = aktuelle_folge
+            laengste_indices = aktuelle_indices
+        aktuelle_folge = []
+        aktuelle_indices = []
+
+    if len(aktuelle_folge) > len(laengste_folge) and lst[-1] != aktuelle_folge[-1]:
+        laengste_folge = aktuelle_folge
+        laengste_indices = aktuelle_indices
+
+    return laengste_folge, laengste_indices
+
+
+all_iters = []
+idx = 0
+for i in range(500):
+    np_data = np.asarray(data)
+    result = roll(np_data, i, Part_2)
+    all_iters.append(result[0])
+    idx+=1
+    print(i)
+print()
+laengste_folge, laengste_indices = finde_laengste_folge(all_iters)
+mod = laengste_indices[-1] - laengste_indices[0] + 1
+result = laengste_folge[(999999999 - laengste_indices[-1]) % mod]
+print("laengste_folge={}, laengste_indices={}".format(laengste_folge, laengste_indices))
+Part_2 = False
+np_data = np.asarray(data)
+print("Part 1: {} \n".format(roll(np_data,1, Part_2)[0]))
+print("Part 2: {}".format(result))
+input("...")