blob: 3d6741d7e2f4f04d036d41947b672cde8289bd20 (
plain) (
blame)
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
|
# Day Fourteen: Docking Data
import std/[os, strutils, sequtils, sugar, tables]
let input: string = paramStr(1)
let program: seq[tuple[address, value: string]] =
input.readFile().strip().split('\n')
.map(param => (param.split(" = ")[0], param.split(" = ")[1]))
var mask: string = 'X'.repeat(36)
var memory: seq[int] = newSeq[int](99999)
for line in program:
if line.address == "mask":
mask = line.value
elif line.address[0..2] == "mem":
var value: string = line.value.parseInt.toBin(36)
for i, bit in mask:
if bit != 'X':
value[i] = bit
memory[line.address[4..^2].parseInt()] = fromBin[int](value)
echo memory.foldl(a + b)
func genadds(addresses: seq[string]): seq[string] =
for address in addresses:
for i, c in address:
if c == 'X':
result.add(address[0..(i-1)] & '0' & address[(i+1)..^1])
result.add(address[0..(i-1)] & '1' & address[(i+1)..^1])
break
if result.len == 0:
return addresses
return genadds(result)
mask = 'X'.repeat(36)
var memorii = initTable[int, int]()
for line in program:
if line.address == "mask":
mask = line.value
elif line.address[0..2] == "mem":
var address = line.address[4..^2].parseInt().toBin(36)
var value = line.value.parseInt.toBin(36)
for i, bit in mask:
if bit == '1':
address[i] = '1'
elif bit == 'X':
address[i] = 'X'
for address in genadds(@[address]):
memorii[fromBin[int](address)] = fromBin[int](value)
echo memorii.values.toSeq.foldl(a + b)
|
