aboutsummaryrefslogtreecommitdiffstats
path: root/src/06/main.zig
blob: b02576bda06f1fa03144e35791f982250a441e47 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
const std = @import("std");
const aoc = @import("aoc");

fn part1(allocator: std.mem.Allocator, input: []u8, args: [][]u8) !?[]u8 {
    _ = args;

    var entryit = std.mem.split(u8, input, "\n\n");
    var answer: u32 = 0;
    while (entryit.next()) |group| {
        var seen = [_]u1{0} ** 256;
        for (group) |c| {
            if (c == ' ' or c == '\n') continue;
            answer += @boolToInt(seen[c] == 0);
            seen[c] = 1;
        }
    }

    return try std.fmt.allocPrint(allocator, "{d}", .{answer});
}

fn part2(allocator: std.mem.Allocator, input: []u8, args: [][]u8) !?[]u8 {
    _ = args;

    var entryit = std.mem.split(u8, input, "\n\n");
    var answer: u32 = 0;
    while (entryit.next()) |group| {
        var count = [_]u16{0} ** 256;
        var members: u16 = 0;
        var memberit = std.mem.tokenize(u8, group, "\n ");
        while (memberit.next()) |member| : (members += 1) {
            for (member) |c| count[c] += 1;
        }
        for (count) |v| answer += @boolToInt(v == members);
    }

    return try std.fmt.allocPrint(allocator, "{d}", .{answer});
}

pub const main = aoc.main(part1, part2, .{ "6748", "3445" });