aboutsummaryrefslogtreecommitdiff
path: root/2022/15-beacon-exclusion-zone/first.zig
blob: 9263efc331aa0d5b5910a0f2766e08567ecb15d5 (plain)
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
const std = @import("std");

const example = @embedFile("example");
const input = @embedFile("input");

var allocator: std.mem.Allocator = undefined;

pub fn main() anyerror!void {
    var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
    defer arena.deinit();
    allocator = arena.allocator();

    try std.testing.expectEqual(try solve(example, 10), 26);
    const result = try solve(input, 2000000);
    try std.io.getStdOut().writer().print("{}\n", .{result});
}

const Interval = struct {
    l: i64,
    r: i64,
};

fn lesserThan(context: void, a: *Interval, b: *Interval) bool {
    _ = context;
    return a.l < b.l;
}

fn solve(puzzle: []const u8, yline: i64) !i64 {
    var it = std.mem.tokenize(u8, puzzle, "\n");
    var intervals = std.ArrayList(*Interval).init(allocator);
    // process input
    while (it.next()) |line| {
        var coords = std.mem.tokenize(u8, line, "Sensor at x=, y=: closest beacon is at x=, y=");
        const sx = try std.fmt.parseInt(i64, coords.next() orelse continue, 10);
        const sy = try std.fmt.parseInt(i64, coords.next() orelse unreachable, 10);
        const bx = try std.fmt.parseInt(i64, coords.next() orelse unreachable, 10);
        const by = try std.fmt.parseInt(i64, coords.next() orelse unreachable, 10);
        const d = try std.math.absInt(sx - bx) + try std.math.absInt(sy - by);
        if (sy - d > yline or sy + d < yline) {
            continue;
        }
        const l = d - try std.math.absInt(yline - sy);
        var i = try allocator.create(Interval);
        i.* = Interval{ .l = sx - l, .r = sx + l };
        try intervals.append(i);
    }
    // sort by left bound
    std.sort.sort(*Interval, intervals.items, {}, lesserThan);
    // Reduce intervals and compute results
    var n: i64 = 0;
    var i: usize = 0;
    while (i < intervals.items.len) : (i += 1) {
        var int = intervals.items[i];
        var j = i + 1;
        while (j < intervals.items.len) {
            const jnt = intervals.items[j];
            if (jnt.l <= int.r) { // jint overlaps
                if (jnt.r > int.r) {
                    int.r = jnt.r;
                }
                _ = intervals.orderedRemove(j);
                continue;
            } else {
                break;
            }
            j += 1;
        }
        n = n + int.r - int.l;
    }
    return n;
}