AdventOfCode2023/day_1/solution.d

58 lines
1.6 KiB
D

module day_1.solution;
import std.file;
import std.string : splitLines;
import std.algorithm;
import std.stdio;
import std.uni : isNumber;
void part1() {
readText("part_1_input.txt")
.splitLines()
.map!((line) {
int a = -1, b = -1;
for (size_t i = 0; i < line.length; i++) {
if (a == -1 && isNumber(line[i])) a = line[i] - '0';
if (b == -1 && isNumber(line[$-i-1])) b = line[$-i-1] - '0';
if (a != -1 && b != -1) break;
}
return a * 10 + b;
})
.sum()
.writeln();
}
void part2() {
int findDigit(string s, size_t idx) {
if (isNumber(s[idx])) return s[idx] - '0';
const string[] words = ["one", "two", "three", "four", "five", "six", "seven", "eight", "nine"];
static foreach (size_t i, string word; words) {
if (s.length - idx >= word.length && s[idx .. idx + word.length] == word) {
return cast(int) i + 1;
}
}
return -1;
}
readText("part_2_input.txt")
.splitLines()
.map!((line) {
int a = -1, b = -1;
for (size_t i = 0; i < line.length; i++) {
int an = findDigit(line, i);
int bn = findDigit(line, line.length-i-1);
if (a == -1 && an != -1) a = an;
if (b == -1 && bn != -1) b = bn;
if (a != -1 && b != -1) break;
}
return a * 10 + b;
})
.sum()
.writeln();
}
void main() {
part1();
part2();
}