This commit is contained in:
2024-12-03 02:40:17 -06:00
parent 398a587d95
commit 012e963b48
4 changed files with 370 additions and 0 deletions

1
.gitignore vendored
View File

@@ -1,4 +1,5 @@
*/.zig-cache
*/zig-out
*/src/test
*/src/test2
*/src/input

66
03/build.zig Normal file
View File

@@ -0,0 +1,66 @@
const std = @import("std");
// Although this function looks imperative, note that its job is to
// declaratively construct a build graph that will be executed by an external
// runner.
pub fn build(b: *std.Build) void {
// Standard target options allows the person running `zig build` to choose
// what target to build for. Here we do not override the defaults, which
// means any target is allowed, and the default is native. Other options
// for restricting supported target set are available.
const target = b.standardTargetOptions(.{});
// Standard optimization options allow the person running `zig build` to select
// between Debug, ReleaseSafe, ReleaseFast, and ReleaseSmall. Here we do not
// set a preferred release mode, allowing the user to decide how to optimize.
const optimize = b.standardOptimizeOption(.{});
const exe = b.addExecutable(.{
.name = "03",
.root_source_file = b.path("src/main.zig"),
.target = target,
.optimize = optimize,
});
// This declares intent for the executable to be installed into the
// standard location when the user invokes the "install" step (the default
// step when running `zig build`).
b.installArtifact(exe);
// This *creates* a Run step in the build graph, to be executed when another
// step is evaluated that depends on it. The next line below will establish
// such a dependency.
const run_cmd = b.addRunArtifact(exe);
// By making the run step depend on the install step, it will be run from the
// installation directory rather than directly from within the cache directory.
// This is not necessary, however, if the application depends on other installed
// files, this ensures they will be present and in the expected location.
run_cmd.step.dependOn(b.getInstallStep());
// This allows the user to pass arguments to the application in the build
// command itself, like this: `zig build run -- arg1 arg2 etc`
if (b.args) |args| {
run_cmd.addArgs(args);
}
// This creates a build step. It will be visible in the `zig build --help` menu,
// and can be selected like this: `zig build run`
// This will evaluate the `run` step rather than the default, which is "install".
const run_step = b.step("run", "Run the app");
run_step.dependOn(&run_cmd.step);
const exe_unit_tests = b.addTest(.{
.root_source_file = b.path("src/main.zig"),
.target = target,
.optimize = optimize,
});
const run_exe_unit_tests = b.addRunArtifact(exe_unit_tests);
// Similar to creating the run step earlier, this exposes a `test` step to
// the `zig build --help` menu, providing a way for the user to request
// running the unit tests.
const test_step = b.step("test", "Run unit tests");
test_step.dependOn(&run_exe_unit_tests.step);
}

72
03/build.zig.zon Normal file
View File

@@ -0,0 +1,72 @@
.{
// This is the default name used by packages depending on this one. For
// example, when a user runs `zig fetch --save <url>`, this field is used
// as the key in the `dependencies` table. Although the user can choose a
// different name, most users will stick with this provided value.
//
// It is redundant to include "zig" in this name because it is already
// within the Zig package namespace.
.name = "03",
// This is a [Semantic Version](https://semver.org/).
// In a future version of Zig it will be used for package deduplication.
.version = "0.0.0",
// This field is optional.
// This is currently advisory only; Zig does not yet do anything
// with this value.
//.minimum_zig_version = "0.11.0",
// This field is optional.
// Each dependency must either provide a `url` and `hash`, or a `path`.
// `zig build --fetch` can be used to fetch all dependencies of a package, recursively.
// Once all dependencies are fetched, `zig build` no longer requires
// internet connectivity.
.dependencies = .{
// See `zig fetch --save <url>` for a command-line interface for adding dependencies.
//.example = .{
// // When updating this field to a new URL, be sure to delete the corresponding
// // `hash`, otherwise you are communicating that you expect to find the old hash at
// // the new URL.
// .url = "https://example.com/foo.tar.gz",
//
// // This is computed from the file contents of the directory of files that is
// // obtained after fetching `url` and applying the inclusion rules given by
// // `paths`.
// //
// // This field is the source of truth; packages do not come from a `url`; they
// // come from a `hash`. `url` is just one of many possible mirrors for how to
// // obtain a package matching this `hash`.
// //
// // Uses the [multihash](https://multiformats.io/multihash/) format.
// .hash = "...",
//
// // When this is provided, the package is found in a directory relative to the
// // build root. In this case the package's hash is irrelevant and therefore not
// // computed. This field and `url` are mutually exclusive.
// .path = "foo",
// // When this is set to `true`, a package is declared to be lazily
// // fetched. This makes the dependency only get fetched if it is
// // actually used.
// .lazy = false,
//},
},
// Specifies the set of files and directories that are included in this package.
// Only files and directories listed here are included in the `hash` that
// is computed for this package. Only files listed here will remain on disk
// when using the zig package manager. As a rule of thumb, one should list
// files required for compilation plus any license(s).
// Paths are relative to the build root. Use the empty string (`""`) to refer to
// the build root itself.
// A directory listed here means that all files within, recursively, are included.
.paths = .{
"build.zig",
"build.zig.zon",
"src",
// For example...
//"LICENSE",
//"README.md",
},
}

231
03/src/main.zig Normal file
View File

@@ -0,0 +1,231 @@
// Day 3! I chose the worst way to do this and then stuck with it
// Also zig is hell I want rust...
const std = @import("std");
var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
const alloc = arena.allocator();
const test_data = @embedFile("test");
const test_data2 = @embedFile("test2");
const input_data = @embedFile("input");
pub fn main() !void {
const stdout = std.io.getStdOut().writer();
const value1 = try run1(input_data);
try stdout.print("{d}\n", .{value1});
const value2 = try run2(input_data);
try stdout.print("{d}\n", .{value2});
}
test "test input part 1" {
const value = try run1(test_data);
std.debug.print("{d}\n", .{value});
try std.testing.expectEqual(value, 161);
}
test "test input part 2" {
const value = try run2(test_data2);
std.debug.print("{d}\n", .{value});
try std.testing.expectEqual(value, 48);
}
const STATE = enum { waiting, m, u, l, open, d, o, n, open_do, close_do, apo, t, open_dont, close_dont, rewind };
fn integerUntilDelimiter(reader: anytype, writer: anytype, delimiter: u8, optional_max_size: ?usize) anyerror!void {
if (optional_max_size) |max_size| {
for (0..max_size) |_| {
const byte: u8 = try reader.readByte();
if (byte == delimiter) return;
if (byte < '0' or byte > '9') return error.BadCharacter;
try writer.writeByte(byte);
}
return error.StreamTooLong;
} else {
while (true) {
const byte: u8 = try reader.readByte();
if (byte == delimiter) return;
try writer.writeByte(byte);
}
// Can not throw `error.StreamTooLong` since there are no boundary.
}
}
fn run2(s: []const u8) !u64 {
// zig doesn't have regex
// sobs
var s_fbs = std.io.fixedBufferStream(s);
var reader = s_fbs.reader();
var state = STATE.waiting;
var buf: [1024]u8 = undefined;
var total: u64 = 0;
var do_mul: bool = true;
// i tried a state machine.
// god i don't know zig at all
// this is so genuinely awful
read: while (true) {
// state transition
state = sw: {
break :sw switch (state) {
STATE.rewind => {
try s_fbs.seekBy(-1);
break :sw STATE.waiting;
},
STATE.waiting => {
switch (reader.readByte() catch {
break :read;
}) {
'm' => {
break :sw STATE.m;
},
'd' => {
break :sw STATE.d;
},
else => {
break :sw STATE.waiting;
},
}
},
STATE.m => if (reader.readByte() catch {
break :read;
} == 'u') STATE.u else STATE.rewind,
STATE.u => if (reader.readByte() catch {
break :read;
} == 'l') STATE.l else STATE.rewind,
STATE.l => if (reader.readByte() catch {
break :read;
} == '(') STATE.open else STATE.rewind,
STATE.d => if (reader.readByte() catch {
break :read;
} == 'o') STATE.o else STATE.rewind,
STATE.o => switch (reader.readByte() catch {
break :read;
}) {
'n' => STATE.n,
'(' => STATE.open_do,
else => STATE.rewind,
},
STATE.open_do => if (reader.readByte() catch {
break :read;
} == ')') STATE.close_do else STATE.rewind,
STATE.n => if (reader.readByte() catch {
break :read;
} == '\'') STATE.apo else STATE.rewind,
STATE.apo => if (reader.readByte() catch {
break :read;
} == 't') STATE.t else STATE.rewind,
STATE.t => if (reader.readByte() catch {
break :read;
} == '(') STATE.open_dont else STATE.rewind,
STATE.open_dont => if (reader.readByte() catch {
break :read;
} == ')') STATE.close_dont else STATE.rewind,
STATE.close_do => {
do_mul = true;
break :sw STATE.waiting;
},
STATE.close_dont => {
do_mul = false;
break :sw STATE.waiting;
},
STATE.open => {
var fbs = std.io.fixedBufferStream(&buf);
integerUntilDelimiter(reader, fbs.writer(), ',', 1024) catch {
state = STATE.waiting;
continue :read;
};
const num0 = std.fmt.parseInt(u64, fbs.getWritten(), 10) catch {
state = STATE.waiting;
continue :read;
};
fbs = std.io.fixedBufferStream(&buf);
integerUntilDelimiter(reader, fbs.writer(), ')', 1024) catch {
state = STATE.waiting;
continue :read;
};
const num1 = std.fmt.parseInt(u64, fbs.getWritten(), 10) catch {
state = STATE.waiting;
continue :read;
};
if (do_mul) {
std.debug.print("{d} {d}\n", .{ num0, num1 });
total += num0 * num1;
}
break :sw STATE.waiting;
},
};
};
}
return total;
}
fn run1(s: []const u8) !u64 {
// zig doesn't have regex
// sobs
var s_fbs = std.io.fixedBufferStream(s);
var reader = s_fbs.reader();
var state = STATE.waiting;
var buf: [1024]u8 = undefined;
var total: u64 = 0;
// i tried a state machine.
// god i don't know zig at all
// this is so genuinely awful
read: while (true) {
// state transition
state = sw: {
break :sw switch (state) {
STATE.waiting => if (reader.readByte() catch {
break :read;
} == 'm') STATE.m else STATE.waiting,
STATE.m => if (reader.readByte() catch {
break :read;
} == 'u') STATE.u else STATE.waiting,
STATE.u => if (reader.readByte() catch {
break :read;
} == 'l') STATE.l else STATE.waiting,
STATE.l => if (reader.readByte() catch {
break :read;
} == '(') STATE.open else STATE.waiting,
STATE.open => {
var fbs = std.io.fixedBufferStream(&buf);
integerUntilDelimiter(reader, fbs.writer(), ',', 1024) catch {
state = STATE.waiting;
continue :read;
};
const num0 = std.fmt.parseInt(u64, fbs.getWritten(), 10) catch {
state = STATE.waiting;
continue :read;
};
fbs = std.io.fixedBufferStream(&buf);
integerUntilDelimiter(reader, fbs.writer(), ')', 1024) catch {
state = STATE.waiting;
continue :read;
};
const num1 = std.fmt.parseInt(u64, fbs.getWritten(), 10) catch {
state = STATE.waiting;
continue :read;
};
std.debug.print("{d} {d}\n", .{ num0, num1 });
total += num0 * num1;
break :sw STATE.waiting;
},
else => STATE.waiting,
};
};
}
return total;
}