Files
advent-of-code/2020/07/part1.js
Nick Thomas f1be11fca8 Add '2020/' from commit 'aaabfa90c9033044d0a9d5fe6776b718711ef46c'
git-subtree-dir: 2020
git-subtree-mainline: ab8f135946
git-subtree-split: aaabfa90c9
2022-01-09 17:06:15 +00:00

50 lines
1.5 KiB
JavaScript

const fs = require('fs');
const OWN_BAG = 'shiny gold';
function parseLine(line) {
let [ self, others ] = line.trim().split(" bags contain ");
if (self === undefined || others === undefined) return [];
if (others == "no other bags.") return []; // [{a: self, b: null, count: 0}];
others = others.replace(".", "").replace(/bags?/g, "").split(",");
return others.map((other) => {
let [ num, ...rest ] = other.trim().split(' ');
return { a: self, b: rest.join(' '), count: Number(num) };
});
}
fs.readFile('input', (err, data) => {
if (err) throw err;
let rules = data.toString().split("\n").map( parseLine ).flat();
let bags = {};
rules.forEach((rule) => {
let parent = bags[rule.a] || {spec: rule.a};
let child = bags[rule.b] || {spec: rule.b};
if (!child.hasOwnProperty('parent')) child.parent = parent;
if (!parent.hasOwnProperty('children')) parent.children = [];
parent.children.push(child);
bags[rule.a] = parent;
bags[rule.b] = child;
});
let tops = Object.values(bags).filter((bag) => !bag.hasOwnProperty('parent') && bag.spec != OWN_BAG);
matchBags = function(bag, parentColours) {
if (bag.spec == OWN_BAG) {
return parentColours;
}
if (!bag.children) return [];
let colours = [...parentColours]; // clone
colours.push(bag.spec);
return bag.children.map((child) => matchBags(child, colours)).flat();
}
let colours = tops.map((top) => matchBags(top, []) ).flat();
console.log( new Set(colours).size );
});