summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--6np.py7
1 files changed, 2 insertions, 5 deletions
diff --git a/6np.py b/6np.py
index 6b97f9c..92273b1 100644
--- a/6np.py
+++ b/6np.py
@@ -3,7 +3,6 @@ from utils import open_day
import numpy as np
fish = Counter(map(int, open_day(6).read().split(',')))
-
fish = np.array([fish[i] for i in range(9)])
mat = np.array([
[0, 0, 0, 0, 0, 0, 1, 0, 1],
@@ -17,7 +16,5 @@ mat = np.array([
[0, 0, 0, 0, 0, 0, 0, 1, 0]
])
-for i in range(256):
- if i == 80: print(sum(fish))
- fish = np.matmul(fish, mat)
-print(sum(fish))
+print(sum(np.matmul(fish, np.linalg.matrix_power(mat, 80))))
+print(sum(np.matmul(fish, np.linalg.matrix_power(mat, 256))))