jdkaplan/advent-of-code
Watch
1
0
Fork
You've already forked advent-of-code
 0
Code Issues Activity

2024: try parseAll

Browse source
This commit is contained in:
Jeremy Kaplan 2024-12-06 23:23:20 -08:00
commit 9e515bfd50
2 changed files with 22 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

17
2024/src/aoc.zig
View file

@ -65,3 +65,20 @@ pub fn AutoHashSet(comptime T: type) type {
} }
}; };
} }
pub fn parseAll(
comptime T: type,
allocator: std.mem.Allocator,
text: []const u8,
sep: []const u8,
) !std.ArrayList(T) {
var all = std.ArrayList(T).init(allocator);
var it = std.mem.tokenizeSequence(u8, text, sep);
while (it.next()) |s| {
const eqn = try T.parse(allocator, s);
try all.append(eqn);
}
return all;
}

28
2024/src/day07.zig
View file

@ -15,7 +15,7 @@ pub fn main() !void {
const text = try file.readToEndAlloc(allocator, comptime 1 << 30); const text = try file.readToEndAlloc(allocator, comptime 1 << 30);
defer allocator.free(text); defer allocator.free(text);
var equations = try Equation.parseAll(allocator, text); var equations = try aoc.parseAll(Equation, allocator, text, "\n");
defer equations.deinit(); defer equations.deinit();
defer { defer {
for (equations.items) |*eqn| eqn.deinit(); for (equations.items) |*eqn| eqn.deinit();
@ -45,7 +45,9 @@ fn part1(allocator: Allocator, equations: []Equation) !u64 {
fn part2(allocator: Allocator, equations: []Equation) !u64 { fn part2(allocator: Allocator, equations: []Equation) !u64 {
var sum: u64 = 0; var sum: u64 = 0;
for (equations) |eqn| { for (equations) |eqn| {
if (try eqn.solvable(allocator, &[_]Operator{ .Plus, .Times }) or try eqn.solvable(allocator, &[_]Operator{ .Plus, .Times, .Concat })) { const ops1 = &[_]Operator{ .Plus, .Times };
const ops2 = &[_]Operator{ .Plus, .Times, .Concat };
if (try eqn.solvable(allocator, ops1) or try eqn.solvable(allocator, ops2)) {
sum += eqn.value; sum += eqn.value;
} }
} }
@ -95,7 +97,7 @@ const Equation = struct {
self.operands.deinit(); self.operands.deinit();
} }
fn parse(allocator: Allocator, line: []const u8) !Self { pub fn parse(allocator: Allocator, line: []const u8) !Self {
const value, const rest = a: { const value, const rest = a: {
var it = std.mem.splitSequence(u8, line, ": "); var it = std.mem.splitSequence(u8, line, ": ");
const value = try std.fmt.parseInt(u64, it.next().?, 10); const value = try std.fmt.parseInt(u64, it.next().?, 10);
@ -115,26 +117,6 @@ const Equation = struct {
return .{ .value = value, .operands = operands }; return .{ .value = value, .operands = operands };
} }
fn parseAll(allocator: Allocator, text: []const u8) !ArrayList(Self) {
var all = ArrayList(Self).init(allocator);
var it = std.mem.tokenizeScalar(u8, text, '\n');
while (it.next()) |line| {
const eqn = try Self.parse(allocator, line);
try all.append(eqn);
}
return all;
}
fn eval(self: Self, operators: ArrayList(Operator)) u64 {
var v = self.operands.items[0];
for (self.operands.items[1 .. operators.items.len + 1], operators.items) |n, op| {
v = op.eval(v, n);
}
return v;
}
fn solvable(self: Self, allocator: Allocator, operators: []const Operator) !bool { fn solvable(self: Self, allocator: Allocator, operators: []const Operator) !bool {
const State = struct { const State = struct {
partial: u64, partial: u64,