71 lines
1.4 KiB
Nix
71 lines
1.4 KiB
Nix
{ pkgs, lib, AoCLib, ... }:
|
|
|
|
with lib;
|
|
|
|
let
|
|
inherit (AoCLib) transformRange chunksOf;
|
|
|
|
compartments = pipe (fileContents ./input.txt) [
|
|
(splitString "\n")
|
|
];
|
|
|
|
splitAtMiddle = s: {
|
|
c1 = substring 0 ((stringLength s) / 2) s;
|
|
c2 = substring ((stringLength s) / 2) (stringLength s) s;
|
|
};
|
|
|
|
charsToSet = s: foldl (set: c: set // { ${c} = true; }) { } (stringToCharacters s);
|
|
|
|
getCommonChar = { c1, c2 }:
|
|
findSingle
|
|
(c: c2.${c} or false)
|
|
"Error: no common chars"
|
|
"Error: multiple chars"
|
|
(attrNames c1);
|
|
|
|
charValue = c: pipe c [
|
|
lib.strings.charToInt
|
|
(transformRange 65 91 (-(65 - 27)))
|
|
(transformRange 97 123 (-(97 - 1)))
|
|
];
|
|
|
|
answer1 = pipe compartments [
|
|
(map splitAtMiddle)
|
|
(map (mapAttrs (_: charsToSet)))
|
|
(map getCommonChar)
|
|
(map charValue)
|
|
(foldr add 0)
|
|
toString
|
|
];
|
|
|
|
toA123 = l: {
|
|
a1 = elemAt l 0;
|
|
a2 = elemAt l 1;
|
|
a3 = elemAt l 2;
|
|
};
|
|
|
|
getCommonChar2 = { a1, a2, a3 }:
|
|
findSingle
|
|
(c: a2.${c} or false && a3.${c} or false)
|
|
"Error: no common chars"
|
|
"Error: multiple chars"
|
|
(attrNames a1);
|
|
|
|
answer2 = pipe compartments [
|
|
(chunksOf 3)
|
|
(map toA123)
|
|
(map (mapAttrs (_: charsToSet)))
|
|
(map getCommonChar2)
|
|
(map charValue)
|
|
(foldr add 0)
|
|
toString
|
|
];
|
|
|
|
in pkgs.writeText "answers" ''
|
|
Task1:
|
|
${answer1}
|
|
|
|
Task2:
|
|
${answer2}
|
|
''
|