-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path25.py
More file actions
80 lines (75 loc) · 2.42 KB
/
25.py
File metadata and controls
80 lines (75 loc) · 2.42 KB
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
print(chr(27)+'[2j')
print('\033c', end='')
# f = open('25.test', 'r')
f = open('25.input', 'r')
lines = [x.strip() for x in f.readlines()]
def solve(registers):
idx = 0
tick = 0
while idx < len(lines):
tick += 1
if tick % 1000000 == 0:
return None
line = lines[idx]
parts = line.split()
# print(tick, idx, parts, registers)
if parts[0] == 'cpy':
if parts[1].isalpha():
registers[parts[2]] = registers[parts[1]]
else:
registers[parts[2]] = int(parts[1])
elif parts[0] == 'mul':
registers[parts[3]] += registers[parts[1]] * registers[parts[2]]
elif parts[0] == 'inc':
registers[parts[1]] = registers[parts[1]] + 1
elif parts[0] == 'out':
# print("OUT", registers[parts[1]])
yield registers[parts[1]]
elif parts[0] == 'dec':
registers[parts[1]] = registers[parts[1]] - 1
elif parts[0] == 'tgl':
tgl_idx = idx + registers[parts[1]]
if tgl_idx < len(lines):
tgl_line = lines[tgl_idx]
tgl_parts = tgl_line.split()
if len(tgl_parts) == 2:
if tgl_parts[0] == 'inc':
tgl_parts[0] = 'dec'
else:
tgl_parts[0] = 'inc'
else:
if tgl_parts[0] == 'jnz':
tgl_parts[0] = 'cpy'
else:
tgl_parts[0] = 'jnz'
lines[tgl_idx] = ' '.join(tgl_parts)
elif parts[0] == 'jnz':
if parts[1].isalpha():
if registers[parts[1]] != 0:
idx += int(parts[2])
continue
else:
if int(parts[1]) != 0:
if parts[2].isalpha():
idx += registers[parts[2]]
else:
idx += int(parts[2])
continue
idx += 1
return registers['a']
test = 0
while True:
count = 0
curr = 1
for x in solve({'a': test, 'b': 0, 'c': 0, 'd': 0}):
# print("OUT", x)
count += 1
if x == curr:
# print("Break")
break
else:
curr = x
if count == 100:
print('Solution:', test)
exit()
test += 1