Compare commits
No commits in common. "24669a60f06bde387fd2d16642f7585849dca060" and "faf81869d60fbff987ae47957b9587cf50b5d041" have entirely different histories.
24669a60f0
...
faf81869d6
185
src/day05.zig
185
src/day05.zig
@ -5,140 +5,117 @@ const mem = std.mem;
|
||||
pub fn main() !void {
|
||||
const input = @embedFile("data/day05.txt");
|
||||
const sln = try solve(util.gpa, input);
|
||||
std.debug.print("{d}\n", .{sln});
|
||||
std.debug.print("{d}\n", .{sln});
|
||||
std.debug.print("{d}\n", .{sln.a});
|
||||
std.debug.print("{d}\n", .{sln.b});
|
||||
}
|
||||
|
||||
fn Range(comptime T: type) type {
|
||||
return struct {
|
||||
start: T,
|
||||
end: T,
|
||||
|
||||
const Self = @This();
|
||||
|
||||
/// Returns the intersection of two ranges.
|
||||
pub fn intersect(self: Self, other: Self) ?Self {
|
||||
const a = if (self.start < other.start) self else other;
|
||||
const b = if (self.start < other.start) other else self;
|
||||
|
||||
if (a.start == a.end or b.start == b.end) return null; // Points don't intersect.
|
||||
|
||||
if (a.end < b.start) return null; // No intersection.
|
||||
if (a.end > b.end) return b; // A is a superset of B.
|
||||
return .{ .start = b.start, .end = a.end };
|
||||
}
|
||||
|
||||
/// Returns the union of two ranges.
|
||||
pub fn join(self: Self, other: Self) ?Self {
|
||||
if (self.intersect(other) == null) return null; // Can't union disjoint ranges.
|
||||
// Now guaranteed that they overlap.
|
||||
return .{ .start = @min(self.start, other.start), .end = @max(self.end, other.end) };
|
||||
}
|
||||
const Solution = struct {
|
||||
a: usize,
|
||||
b: usize,
|
||||
};
|
||||
|
||||
const Transform = struct {
|
||||
from: []const u8,
|
||||
to: []const u8,
|
||||
};
|
||||
}
|
||||
|
||||
const Map = struct {
|
||||
offset: isize,
|
||||
source: Range(usize),
|
||||
dest_start: usize,
|
||||
source_start: usize,
|
||||
range: usize,
|
||||
};
|
||||
|
||||
const Component = struct {
|
||||
stage: usize,
|
||||
range: Range(usize),
|
||||
name: []const u8,
|
||||
value: usize,
|
||||
};
|
||||
|
||||
fn mergeRanges(ranges: *std.ArrayList(Component)) void {
|
||||
var i: usize = 0;
|
||||
while (i < ranges.items.len) : (i += 1) {
|
||||
var item = ranges.items[i];
|
||||
var j = i + 1;
|
||||
while (j < ranges.items.len) {
|
||||
const next = ranges.items[j];
|
||||
if (item.range.join(next.range)) |merged| {
|
||||
const merged_item = .{ .stage = item.stage, .range = merged };
|
||||
item = merged_item;
|
||||
ranges.items[i] = merged_item;
|
||||
_ = ranges.swapRemove(j);
|
||||
} else {
|
||||
j += 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
const UsizeSet = std.AutoHashMap(usize, void);
|
||||
|
||||
fn solve(alloc: mem.Allocator, input: []const u8) !usize {
|
||||
fn solve(alloc: mem.Allocator, input: []const u8) !Solution {
|
||||
var it = util.splitLines(input);
|
||||
const seed_line = it.next().?;
|
||||
const seeds = try util.parseIntsScalar(usize, alloc, seed_line[mem.indexOfScalar(u8, seed_line, ':').? + 1 ..], .{});
|
||||
defer alloc.free(seeds);
|
||||
|
||||
var ranges = std.ArrayList(Component).init(alloc);
|
||||
defer ranges.deinit();
|
||||
var last_values = UsizeSet.init(alloc);
|
||||
var working_values = UsizeSet.init(alloc);
|
||||
defer last_values.deinit();
|
||||
defer working_values.deinit();
|
||||
|
||||
{
|
||||
var index: usize = 0;
|
||||
while (index < seeds.len) : (index += 2) {
|
||||
try ranges.append(.{
|
||||
.stage = 0,
|
||||
.range = .{
|
||||
.start = seeds[index],
|
||||
.end = seeds[index] + seeds[index + 1],
|
||||
},
|
||||
});
|
||||
}
|
||||
var low_loc: usize = std.math.maxInt(usize);
|
||||
|
||||
var seeds_index: usize = 0;
|
||||
while (seeds_index < seeds.len) : (seeds_index += 2) {
|
||||
last_values.clearRetainingCapacity();
|
||||
working_values.clearRetainingCapacity();
|
||||
|
||||
std.debug.print("= seeding {d}/{d} =\n", .{ seeds_index, seeds.len });
|
||||
const start = seeds[seeds_index];
|
||||
const range = seeds[seeds_index + 1];
|
||||
for (start..start + range) |seed| {
|
||||
// std.debug.print("seed {d}\n", .{seed});
|
||||
try working_values.put(seed, {});
|
||||
}
|
||||
|
||||
var current_stage: usize = 0;
|
||||
// Reset parse input.
|
||||
it = util.splitLines(input);
|
||||
_ = it.next().?;
|
||||
var current_component: []const u8 = undefined;
|
||||
while (it.next()) |line| {
|
||||
if (line.len == 0) {
|
||||
for (ranges.items, 0..) |item, index| {
|
||||
if (item.stage != current_stage) {
|
||||
ranges.items[index] = .{ .stage = current_stage, .range = item.range };
|
||||
std.debug.print("== swap ==\n", .{});
|
||||
// Copy all remaining
|
||||
var lv_it = last_values.iterator();
|
||||
while (lv_it.next()) |entry| {
|
||||
try working_values.put(entry.key_ptr.*, {});
|
||||
}
|
||||
}
|
||||
mergeRanges(&ranges);
|
||||
// Swap working sets
|
||||
mem.swap(UsizeSet, &last_values, &working_values);
|
||||
working_values.clearRetainingCapacity();
|
||||
} else if (mem.endsWith(u8, line, "map:")) {
|
||||
current_stage += 1;
|
||||
var t_it = util.split(line, ' ');
|
||||
var comp_it = util.split(t_it.next().?, '-');
|
||||
_ = comp_it.next(); // A
|
||||
_ = comp_it.next(); // to
|
||||
current_component = comp_it.next().?; // B
|
||||
} else {
|
||||
const map_line = try util.parseIntsScalar(usize, alloc, line, .{});
|
||||
defer alloc.free(map_line);
|
||||
const map = Map{
|
||||
.offset = @as(isize, @intCast(map_line[0])) - @as(isize, @intCast(map_line[1])),
|
||||
.source = .{ .start = map_line[1], .end = map_line[1] + map_line[2] },
|
||||
};
|
||||
std.debug.print("== {s} {d}/{d}\n", .{ current_component, working_values.count(), last_values.count() });
|
||||
const map = try util.parseIntsScalar(usize, alloc, line, .{});
|
||||
defer alloc.free(map);
|
||||
|
||||
var i: usize = ranges.items.len;
|
||||
while (i != 0) : (i -= 1) {
|
||||
if (ranges.items[i - 1].stage == current_stage) continue;
|
||||
const item = ranges.swapRemove(i - 1);
|
||||
|
||||
if (item.range.intersect(.{ .start = 0, .end = map.source.start })) |section| {
|
||||
// Non-transitioning section.
|
||||
try ranges.append(.{ .stage = item.stage, .range = section });
|
||||
var to_delete = UsizeSet.init(alloc);
|
||||
defer to_delete.deinit();
|
||||
var lv_it = last_values.iterator();
|
||||
while (lv_it.next()) |entry| {
|
||||
// if (transitioned[index]) {
|
||||
// std.debug.print("{d} already mapped\n", .{index});
|
||||
// continue :map_blk;
|
||||
// }
|
||||
const item = entry.key_ptr.*;
|
||||
if (map[1] <= item and item <= map[1] + map[2]) {
|
||||
const offset = @as(isize, @intCast(map[0])) - @as(isize, @intCast(map[1]));
|
||||
const res: usize = @intCast(@as(isize, @intCast(item)) + offset);
|
||||
// std.debug.print("{d} => {d}\n", .{ item, res });
|
||||
try working_values.put(res, {});
|
||||
try to_delete.put(item, {});
|
||||
}
|
||||
if (item.range.intersect(map.source)) |section| {
|
||||
// Transitioning section.
|
||||
try ranges.append(.{
|
||||
.stage = current_stage,
|
||||
.range = .{
|
||||
.start = @intCast(@as(isize, @intCast(section.start)) + map.offset),
|
||||
.end = @intCast(@as(isize, @intCast(section.end)) + map.offset),
|
||||
},
|
||||
});
|
||||
}
|
||||
if (item.range.intersect(.{ .start = map.source.end, .end = std.math.maxInt(usize) })) |section| {
|
||||
// Non-transitioning section.
|
||||
try ranges.append(.{ .stage = item.stage, .range = section });
|
||||
}
|
||||
var td_it = to_delete.iterator();
|
||||
while (td_it.next()) |entry| {
|
||||
// std.debug.print("pop {d}\n", .{entry.key_ptr.*});
|
||||
_ = last_values.remove(entry.key_ptr.*);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
var min_loc: usize = std.math.maxInt(usize);
|
||||
for (ranges.items) |item| {
|
||||
min_loc = @min(min_loc, item.range.start);
|
||||
var lv_it = last_values.iterator();
|
||||
while (lv_it.next()) |value| {
|
||||
low_loc = @min(low_loc, value.key_ptr.*);
|
||||
}
|
||||
return min_loc;
|
||||
}
|
||||
|
||||
return .{ .a = low_loc, .b = 0 };
|
||||
}
|
||||
|
||||
test "silver" {
|
||||
@ -179,7 +156,7 @@ test "silver" {
|
||||
;
|
||||
_ = input;
|
||||
// const sln = try solve(std.testing.allocator, input);
|
||||
// try std.testing.expectEqual(@as(usize, 35), sln);
|
||||
// try std.testing.expectEqual(@as(usize, 35), sln.a);
|
||||
}
|
||||
|
||||
test "gold" {
|
||||
@ -219,5 +196,5 @@ test "gold" {
|
||||
\\56 93 4
|
||||
;
|
||||
const sln = try solve(std.testing.allocator, input);
|
||||
try std.testing.expectEqual(@as(usize, 46), sln);
|
||||
try std.testing.expectEqual(@as(usize, 46), sln.a);
|
||||
}
|
||||
|
@ -1,85 +0,0 @@
|
||||
const std = @import("std");
|
||||
const util = @import("util.zig");
|
||||
const mem = std.mem;
|
||||
|
||||
pub fn main() !void {
|
||||
const input = @embedFile("data/day06.txt");
|
||||
const sln = try solve(util.gpa, input);
|
||||
std.debug.print("{d}\n", .{sln.a});
|
||||
std.debug.print("{d}\n", .{sln.b});
|
||||
}
|
||||
|
||||
const Solution = struct {
|
||||
a: usize,
|
||||
b: usize,
|
||||
};
|
||||
|
||||
const Race = struct {
|
||||
time: usize,
|
||||
distance: usize,
|
||||
};
|
||||
|
||||
fn solve(alloc: mem.Allocator, input: []const u8) !Solution {
|
||||
const races: []Race = r: {
|
||||
var it = util.splitLines(input);
|
||||
const times = try util.parseIntsScalar(usize, alloc, it.next().?[10..], .{});
|
||||
const distances = try util.parseIntsScalar(usize, alloc, it.next().?[10..], .{});
|
||||
defer alloc.free(times);
|
||||
defer alloc.free(distances);
|
||||
var al = try std.ArrayList(Race).initCapacity(alloc, times.len);
|
||||
for (times, distances) |t, d| {
|
||||
try al.append(.{ .time = t, .distance = d });
|
||||
}
|
||||
break :r try al.toOwnedSlice();
|
||||
};
|
||||
defer alloc.free(races);
|
||||
|
||||
const bigRace: Race = r: {
|
||||
var it = util.splitLines(input);
|
||||
var vals: [2]usize = undefined;
|
||||
for (0..vals.len) |i| {
|
||||
const line = try mem.replaceOwned(u8, alloc, it.next().?[10..], " ", "");
|
||||
defer alloc.free(line);
|
||||
vals[i] = try std.fmt.parseInt(usize, line, 10);
|
||||
}
|
||||
break :r .{ .time = vals[0], .distance = vals[1] };
|
||||
};
|
||||
|
||||
var result: usize = 1;
|
||||
for (races) |race| {
|
||||
var winning: usize = 0;
|
||||
for (0..race.time - 1) |speed| {
|
||||
const distance = speed * (race.time - speed);
|
||||
if (distance > race.distance) winning += 1;
|
||||
}
|
||||
result *= winning;
|
||||
}
|
||||
|
||||
var result2: usize = 1;
|
||||
var winning2: usize = 0;
|
||||
for (0..bigRace.time - 1) |speed| {
|
||||
const distance = speed * (bigRace.time - speed);
|
||||
if (distance > bigRace.distance) winning2 += 1;
|
||||
}
|
||||
result2 *= winning2;
|
||||
|
||||
return .{ .a = result, .b = result2 };
|
||||
}
|
||||
|
||||
test "silver" {
|
||||
const input =
|
||||
\\Time: 7 15 30
|
||||
\\Distance: 9 40 200
|
||||
;
|
||||
const sln = try solve(std.testing.allocator, input);
|
||||
try std.testing.expectEqual(@as(usize, 288), sln.a);
|
||||
}
|
||||
|
||||
test "gold" {
|
||||
const input =
|
||||
\\Time: 7 15 30
|
||||
\\Distance: 9 40 200
|
||||
;
|
||||
const sln = try solve(std.testing.allocator, input);
|
||||
try std.testing.expectEqual(@as(usize, 288), sln.b);
|
||||
}
|
Loading…
Reference in New Issue
Block a user