mirror of
https://github.com/HenkKalkwater/aoc-2020
synced 2024-11-22 11:05:18 +00:00
39 lines
923 B
D
39 lines
923 B
D
import std.array;
|
|
import std.algorithm;
|
|
import std.conv;
|
|
import std.range;
|
|
import std.stdio;
|
|
import std.variant;
|
|
|
|
import dayutil;
|
|
|
|
Variant run(int part, File input, bool bigboy, string[] args) {
|
|
uint[] numbers = input.byLineCopy.map!(to!uint).array ~ [0u];
|
|
auto sortedNumbers = numbers.sort;
|
|
|
|
Variant result = parts!size_t(part,
|
|
() => part1(sortedNumbers));
|
|
return result;
|
|
}
|
|
|
|
size_t part1(T)(SortedRange!(T[]) numbers) {
|
|
size_t gap1 = 0, gap3 = 0;
|
|
|
|
foreach(window; numbers.slide!(No.withPartial)(2)) {
|
|
size_t diff = window[1] - window[0];
|
|
if (diff == 3) gap3++;
|
|
if (diff == 1) gap1++;
|
|
}
|
|
// Device adapter is always rated 3 volts higher
|
|
gap3++;
|
|
writeln(gap1, ", ", gap3);
|
|
return gap1 * gap3;
|
|
}
|
|
|
|
unittest {
|
|
uint[] numbers = [28, 33, 18, 42, 31, 14, 46, 20, 48, 47, 24, 23, 49, 45, 19, 38, 39, 11, 1, 32,
|
|
25, 35, 8, 17, 7, 9, 4, 2, 34, 10, 3];
|
|
numbers ~= [0u];
|
|
assert(part1(numbers.sort) == 220);
|
|
}
|