-
Notifications
You must be signed in to change notification settings - Fork 0
/
day14part2.py
41 lines (38 loc) · 1.37 KB
/
day14part2.py
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
import sys
mask = None
mem = {}
for line in sys.stdin:
item, value = line[: -1].split(' = ')
if item == 'mask':
mask = list(value)
else:
addr = list(bin(int(item[: -1].split('[')[1]))[2:])
addr = ['0'] * (len(mask) - len(addr)) + addr
value = int(value)
addresses = [[]]
for i in range(len(addr)):
print(i)
tmp_addresses = []
if mask[i] == '0':
for address in addresses:
address_copy = address.copy()
address_copy.append(addr[i])
tmp_addresses.append(address_copy)
elif mask[i] == '1':
for address in addresses:
address_copy = address.copy()
address_copy.append('1')
tmp_addresses.append(address_copy)
else:
for address in addresses:
address_copy = address.copy()
address_copy.append('0')
tmp_addresses.append(address_copy)
address_copy = address.copy()
address_copy.append('1')
tmp_addresses.append(address_copy)
addresses = tmp_addresses
for address in addresses:
address = int(''.join(address))
mem[address] = value
print(sum(mem.values()))