1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
import * as fs from 'fs';
const setMask = /mask = ([10X]+)/;
const operation = /mem\[(\d+)] = (\d+)/;
function applyMask(addr, mask) {
let ret = [parseInt(addr).toString(2).padStart(36, '0')];
for (let i=0; i<36; ++i) {
switch (mask.charAt(i)) {
case '1':
ret = ret.map(a => a.substring(0, i) + '1' + a.substring(i+1));
break;
case 'X':
ret = ret.reduce((acc, a) => acc.concat([
a.substring(0, i) + '1' + a.substring(i+1),
a.substring(0, i) + '0' + a.substring(i+1),
]), []);
}
}
return ret.map(a => parseInt(a, 2));
}
function processMasks(lines) {
let ret = new Map();
let mask = '';
lines.forEach(line => {
const m = line.match(setMask);
if (m) {
mask = m[1];
} else {
const o = line.match(operation);
const addr = o[1];
const value = parseInt(o[2]);
applyMask(addr, mask).forEach(r => ret.set(parseInt(r), value));
}
});
return ret;
}
function load(filename) {
return processMasks(fs.readFileSync(filename, 'utf8')
.trim()
.split('\n')
);
}
let example = load('example2');
let input = load('input');
function solve(input) {
let sum = 0;
input.forEach(value => sum += value);
return sum;
}
const exampleOutput = solve(example);
if (exampleOutput !== 208) {
console.log('Example failed with ' + exampleOutput);
process.exit(1); // eslint-disable-line
}
console.log(solve(input));
|
