From 231cd753962156adb2ba42e198b4b13f84fadfb4 Mon Sep 17 00:00:00 2001
From: Nguyễn Gia Phong <mcsinyx@disroot.org>
Date: Mon, 6 Dec 2021 15:09:46 +0700
Subject: [aoc] Add day 6

---
 aoc/2021/06/part-one.py  |  9 +++++++++
 aoc/2021/06/part-two.zig | 22 ++++++++++++++++++++++
 2 files changed, 31 insertions(+)
 create mode 100644 aoc/2021/06/part-one.py
 create mode 100644 aoc/2021/06/part-two.zig

(limited to 'aoc')

diff --git a/aoc/2021/06/part-one.py b/aoc/2021/06/part-one.py
new file mode 100644
index 0000000..f1ad2fb
--- /dev/null
+++ b/aoc/2021/06/part-one.py
@@ -0,0 +1,9 @@
+from collections import deque
+from sys import stdin
+
+timer = deque([0] * 9)
+for i in map(int, stdin.read().strip().split(',')): timer[i] += 1
+for i in range(80):
+    timer.rotate(-1)
+    timer[6] += timer[-1]
+print(sum(timer))
diff --git a/aoc/2021/06/part-two.zig b/aoc/2021/06/part-two.zig
new file mode 100644
index 0000000..72bf877
--- /dev/null
+++ b/aoc/2021/06/part-two.zig
@@ -0,0 +1,22 @@
+const parseUnsigned = std.fmt.parseUnsigned;
+const print = std.debug.print;
+const rotate = std.mem.rotate;
+const std = @import("std");
+const tokenize = std.mem.tokenize;
+
+const input = @embedFile("input");
+
+pub fn main() !void {
+    var timer = [_]usize{ 0 } ** 9;
+    defer print("{}\n", .{ @reduce(.Add, @as(@Vector(9, usize), timer)) });
+
+    var list = tokenize(input[0..input.len-1], ",");
+    while (list.next()) |i|
+        timer[try parseUnsigned(usize, i, 10)] += 1;
+
+    var i: usize = 0;
+    while (i < 256) : (i += 1) {
+        rotate(usize, timer[0..], 1);
+        timer[6] += timer[8];
+    }
+}
-- 
cgit 1.4.1