2022-07 in zig

2022/07-no-space-left-on-device/first.zig

@@ -0,0 +1,55 @@
+const std = @import("std");
+
+const example = @embedFile("example");
+const input = @embedFile("input");
+
+pub fn main() anyerror!void {
+    var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
+    defer arena.deinit();
+    const allocator = arena.allocator();
+
+    try std.testing.expectEqual(try solve(example, allocator), 95437);
+    const result = try solve(input, allocator);
+    try std.io.getStdOut().writer().print("{}\n", .{result});
+}
+
+fn solve(puzzle: []const u8, allocator: std.mem.Allocator) !u64 {
+    var it = std.mem.tokenize(u8, puzzle, "\n");
+    var tot: usize = 0;
+    var n: usize = 0; // depth
+    var stack = std.ArrayList(u64).init(allocator);
+    _ = it.next() orelse unreachable; // drop the cd /
+    try stack.append(0);
+    while (it.next()) |line| {
+        if (line[2] == 'c') { // a cd command
+            if (line[5] == '.') { // going up
+                const v = stack.pop();
+                n -= 1;
+                stack.items[n] += v;
+                if (v <= 100000) {
+                    tot += v;
+                }
+            } else { // going down
+                n += 1;
+                try stack.append(0);
+            }
+        } else if (line[2] == 'l') { // a ls command
+            continue;
+        } else if (line[2] == 'r') { // a dir entry in a ls command
+            continue;
+        } else { // a file entry in a ls command
+            var elts = std.mem.split(u8, line, " ");
+            const v = try std.fmt.parseInt(u64, elts.next() orelse unreachable, 10);
+            stack.items[n] += v;
+        }
+    }
+    while (n > 1) : (n -= 1) {
+        const v = stack.pop();
+        n -= 1;
+        stack.items[n] += v;
+        if (v <= 100000) {
+            tot += v;
+        }
+    }
+    return tot;
+}