Update day 6 solution

day-06/day-06.py

@@ -3,40 +3,26 @@ from pathlib import Path
 
 
 def part_1(input):
-    result = 0
+    numbers = [int(n) for n in input[0].split(',')]
-    numbers = input[0].split(',')
+    count = [len([x for x in numbers if x == i]) for i in range(9)]
-    numbers = [int(n) for n in numbers]
+    for _ in range(80):
-    for i in range(80):
+        tmp = count[0]
-        new_numbers = []
+        count = [count[i+1] for i in range(8)]
-        new = 0
+        count[6] += tmp
-        for n in numbers:
+        count[8] = tmp
-            if n > 0:
+    result = sum(count)
-                n -= 1
-            else:
-                n = 6
-                new += 1
-            new_numbers.append(n)
-        new_numbers += [8 for x in range(new)]
-        numbers = new_numbers
-    result = len(numbers)
     print("Part 1 result:", result)
 
 
 def part_2(input):
-    result = 0
+    numbers = [int(n) for n in input[0].split(',')]
-    numbers = input[0].split(',')
+    count = [len([x for x in numbers if x == i]) for i in range(9)]
-    numbers = [int(n) for n in numbers]
+    for _ in range(256):
-    count = [0] * 9
-    for i in range(9):
-        count[i] = len([x for x in numbers if x == i])
-    for i in range(256):
         tmp = count[0]
-        for j in range(8):
+        count = [count[i+1] for i in range(8)]
-            count[j] = count[j+1]
-        count[8] = tmp
         count[6] += tmp
-    for c in count:
+        count[8] = tmp
-        result += c
+    result = sum(count)
     print("Part 2 result:", result)