diff options
Diffstat (limited to 'src/06/main.zig')
| -rw-r--r-- | src/06/main.zig | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/src/06/main.zig b/src/06/main.zig new file mode 100644 index 0000000..b02576b --- /dev/null +++ b/src/06/main.zig @@ -0,0 +1,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" }); |
