adventofcode/2020/8b.py at master · Programator2/adventofcode · GitHub

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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
from aocd import submit
from aoc import *


def submitb(answer):
    submit(answer, part='b', day=8, year=2020)


FILE = '8_test.txt'
FILE = '8.txt'


def main():
    inp = lines(FILE)
    acc = 0
    ins = set([0])
    ip = 0
    nops = []
    jmps = []
    for n, l in enumerate(inp):
        if inp[n][:3] == 'nop':
            nops.append(n)
        elif inp[n][:3] == 'jmp':
            jmps.append(n)
    terminated = False
    for n in nops:
        ins = set([0])
        ip = 0
        acc = 0
        inp[n] = 'jmp ' + inp[n][4:]
        while True:
            torun = inp[ip][:3]
            print(torun)
            if torun == 'nop':
                ip += 1
                if ip == len(inp):
                    terminated = True
                    break
                ip %= len(inp)
                if ip in ins:
                    break
                ins.add(ip)
            elif torun == 'jmp':
                ip += int(inp[ip][4:])
                if ip == len(inp):
                    terminated = True
                    break
                ip %= len(inp)
                if ip in ins:
                    break
                ins.add(ip)
            elif torun == 'acc':
                acc += int(inp[ip][4:])
                ip += 1
                if ip == len(inp):
                    terminated = True
                    break
                ip %= len(inp)
                if ip in ins:
                    break
                ins.add(ip)
        inp[n] = 'nop ' + inp[n][4:]
        if terminated:
            break
    if not terminated:
        for n in jmps:
            ins = set([0])
            ip = 0
            acc = 0
            inp[n] = 'nop ' + inp[n][4:]
            while True:
                torun = inp[ip][:3]
                print(torun)
                if torun == 'nop':
                    ip += 1
                    if ip == len(inp):
                        terminated = True
                        break
                    ip %= len(inp)
                    if ip in ins:
                        break
                    ins.add(ip)
                elif torun == 'jmp':
                    ip += int(inp[ip][4:])
                    if ip == len(inp):
                        terminated = True
                        break
                    ip %= len(inp)
                    if ip in ins:
                        break
                    ins.add(ip)
                elif torun == 'acc':
                    acc += int(inp[ip][4:])
                    ip += 1
                    if ip == len(inp):
                        terminated = True
                        break
                    ip %= len(inp)
                    if ip in ins:
                        break
                    ins.add(ip)
            inp[n] = 'jmp ' + inp[n][4:]
            if terminated:
                break
    out = acc
    print(out)
    input()
    print('submitting')
    submitb(out)


main()