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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
const std = @import("std");
const print = std.debug.print;
pub fn solve(part: []u8, reader: *std.Io.Reader, allocator: std.mem.Allocator) !void {
if (std.mem.eql(u8, part, "1")) {
try part1(reader, allocator);
} else {
try part2(reader, allocator);
}
}
const Range = struct { start: usize, end: usize };
fn part1(reader: *std.Io.Reader, allocator: std.mem.Allocator) !void {
var output: usize = 0;
var ranges = try std.ArrayList(Range).initCapacity(allocator, 0);
defer ranges.deinit(allocator);
while (reader.takeDelimiter('\n')) |line| {
if (line == null or line.?.len == 0){
break;
}
var split = std.mem.splitScalar(u8, line.?, '-');
const start = try std.fmt.parseInt(usize, split.next().?, 10);
const end = try std.fmt.parseInt(usize, split.next().?, 10);
try ranges.append(allocator, Range{ .start=start, .end=end});
} else |_| {}
while (reader.takeDelimiter('\n')) |line| {
if (line == null) {
break;
}
const id = try std.fmt.parseInt(usize, line.?, 10);
for(ranges.items) |range| {
if (id >= range.start and id <= range.end) {
output += 1;
break;
}
}
} else |_| {}
print("{d}\n", .{output});
}
fn cmp(_: void, a: Range, b: Range) bool {
return a.start < b.start;
}
fn part2(reader: *std.Io.Reader, allocator: std.mem.Allocator) !void {
var output: usize = 0;
var ranges = try std.ArrayList(Range).initCapacity(allocator, 0);
defer ranges.deinit(allocator);
while (reader.takeDelimiter('\n')) |line| {
if (line == null or line.?.len == 0){
break;
}
var split = std.mem.splitScalar(u8, line.?, '-');
const start = try std.fmt.parseInt(usize, split.next().?, 10);
const end = try std.fmt.parseInt(usize, split.next().?, 10);
try ranges.append(allocator, Range{ .start=start, .end=end});
} else |_| {}
std.mem.sort(Range,ranges.items, {}, cmp);
var more = try std.ArrayList(Range).initCapacity(allocator, 0);
defer more.deinit(allocator);
outer: for (ranges.items) |r| {
const start = r.start;
const end = r.end;
for (more.items) |*range| {
if (start >= range.start and start <= range.end) {
if (end > range.end){
range.end = end;
}
continue :outer;
} else if (start < range.start and end >= range.start) {
range.start = start;
continue :outer;
}
}
try more.append(allocator, Range{ .start=start, .end=end});
}
for (more.items) |range| {
output += range.end - range.start + 1;
}
print("{d}\n", .{output});
}
|
