-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path33.py
62 lines (61 loc) · 1.33 KB
/
33.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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
f = open("33.in")
a = 0
b = 0
c = 0
prog = []
for line in f.readlines():
if "A" in line:
a = int(line.removeprefix("Register A: "))
if "B" in line:
b = int(line.removeprefix("Register B: "))
if "C" in line:
c = int(line.removeprefix("Register C: "))
if "P" in line:
prog = list(map(int, line.removeprefix("Program: ").split(",")))
output = []
pointer = 0
while pointer < len(prog):
inst = prog[pointer]
combo = prog[pointer+1]
literal = prog[pointer+1]
if combo == 4:
combo = a
elif combo == 5:
combo = b
elif combo == 6:
combo = c
if inst == 0:
a = int(a / 2**combo)
pointer += 2
continue
if inst == 1:
b = b ^ literal
pointer += 2
continue
if inst == 2:
b = combo % 8
pointer += 2
continue
if inst == 3:
if a != 0:
pointer = literal
continue
pointer += 2
continue
if inst == 4:
b = b ^ c
pointer += 2
continue
if inst == 5:
output.append(combo % 8)
pointer += 2
continue
if inst == 6:
b = int(a / 2**combo)
pointer += 2
continue
if inst == 7:
c = int(a / 2**combo)
pointer += 2
continue
print(",".join(map(str, output)))