From 231cd753962156adb2ba42e198b4b13f84fadfb4 Mon Sep 17 00:00:00 2001 From: Nguyễn Gia Phong 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