summaryrefslogtreecommitdiffstats
path: root/6np.py
diff options
context:
space:
mode:
Diffstat (limited to '6np.py')
-rw-r--r--6np.py23
1 files changed, 23 insertions, 0 deletions
diff --git a/6np.py b/6np.py
new file mode 100644
index 0000000..6b97f9c
--- /dev/null
+++ b/6np.py
@@ -0,0 +1,23 @@
+from collections import Counter
+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],
+ [1, 0, 0, 0, 0, 0, 0, 0, 0],
+ [0, 1, 0, 0, 0, 0, 0, 0, 0],
+ [0, 0, 1, 0, 0, 0, 0, 0, 0],
+ [0, 0, 0, 1, 0, 0, 0, 0, 0],
+ [0, 0, 0, 0, 1, 0, 0, 0, 0],
+ [0, 0, 0, 0, 0, 1, 0, 0, 0],
+ [0, 0, 0, 0, 0, 0, 1, 0, 0],
+ [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))