42 lines
1020 B
TypeScript
42 lines
1020 B
TypeScript
|
import { readData } from "./solve_1.ts";
|
||
|
|
|
||
|
|
const memo = new Map<string, number>();
|
||
|
|
|
||
|
|
export function countAvailableWords(pieces: Set<string>, word: string): number {
|
||
|
|
if (memo.has(word)) {
|
||
|
|
return memo.get(word)!;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (word.length === 0) {
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
if (pieces.size === 0) {
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
let count = 0;
|
||
|
|
for (const p of pieces) {
|
||
|
|
if (word.startsWith(p)) {
|
||
|
|
const newWord = word.slice(p.length);
|
||
|
|
const newPieces = pieces;
|
||
|
|
const c = countAvailableWords(newPieces, newWord);
|
||
|
|
count += c;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
memo.set(word, count);
|
||
|
|
return count;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (import.meta.main) {
|
||
|
|
const problem = await readData("input.txt");
|
||
|
|
console.log(problem);
|
||
|
|
const pieces = new Set(problem.piece);
|
||
|
|
let count = 0;
|
||
|
|
for (const word of problem.words) {
|
||
|
|
const c = countAvailableWords(pieces, word);
|
||
|
|
console.log(`word: ${word}, count: ${c}`);
|
||
|
|
count += c;
|
||
|
|
}
|
||
|
|
console.log(count);
|
||
|
|
}
|