CSES - Aalto Competitive Programming 2026 - wk3 - Mon

Submission details
Task:	Wheel of fortune
Sender:	aalto26cm_018
Submission time:	2026-09-14 17:49:11 +0300
Language:	Python3 (PyPy3)
Status:	READY
Result:	TIME LIMIT EXCEEDED

Test results
test	verdict	time
#1	ACCEPTED	0.04 s	details
#2	ACCEPTED	0.04 s	details
#3	ACCEPTED	0.04 s	details
#4	ACCEPTED	0.04 s	details
#5	ACCEPTED	0.04 s	details
#6	ACCEPTED	0.06 s	details
#7	ACCEPTED	0.04 s	details
#8	ACCEPTED	0.04 s	details
#9	ACCEPTED	0.04 s	details
#10	ACCEPTED	0.04 s	details
#11	ACCEPTED	0.04 s	details
#12	ACCEPTED	0.06 s	details
#13	ACCEPTED	0.04 s	details
#14	ACCEPTED	0.04 s	details
#15	ACCEPTED	0.04 s	details
#16	ACCEPTED	0.04 s	details
#17	ACCEPTED	0.04 s	details
#18	ACCEPTED	0.06 s	details
#19	ACCEPTED	0.04 s	details
#20	ACCEPTED	0.04 s	details
#21	ACCEPTED	0.04 s	details
#22	ACCEPTED	0.04 s	details
#23	ACCEPTED	0.04 s	details
#24	ACCEPTED	0.06 s	details
#25	TIME LIMIT EXCEEDED	--	details
#26	TIME LIMIT EXCEEDED	--	details
#27	TIME LIMIT EXCEEDED	--	details
#28	TIME LIMIT EXCEEDED	--	details
#29	TIME LIMIT EXCEEDED	--	details
#30	TIME LIMIT EXCEEDED	--	details
#31	TIME LIMIT EXCEEDED	--	details
#32	TIME LIMIT EXCEEDED	--	details
#33	RUNTIME ERROR	0.11 s	details
#34	RUNTIME ERROR	0.11 s	details
#35	RUNTIME ERROR	0.11 s	details
#36	RUNTIME ERROR	0.11 s	details
#37	RUNTIME ERROR	0.12 s	details
#38	ACCEPTED	0.04 s	details
#39	ACCEPTED	0.04 s	details
#40	ACCEPTED	0.04 s	details
#41	ACCEPTED	0.04 s	details
#42	ACCEPTED	0.04 s	details
#43	ACCEPTED	0.06 s	details
#44	ACCEPTED	0.04 s	details
#45	ACCEPTED	0.04 s	details
#46	ACCEPTED	0.04 s	details
#47	ACCEPTED	0.04 s	details
#48	ACCEPTED	0.04 s	details
#49	ACCEPTED	0.06 s	details
#50	ACCEPTED	0.04 s	details
#51	ACCEPTED	0.04 s	details
#52	ACCEPTED	0.04 s	details
#53	ACCEPTED	0.04 s	details
#54	ACCEPTED	0.04 s	details
#55	ACCEPTED	0.06 s	details
#56	ACCEPTED	0.04 s	details
#57	ACCEPTED	0.04 s	details
#58	ACCEPTED	0.04 s	details
#59	ACCEPTED	0.04 s	details
#60	ACCEPTED	0.04 s	details
#61	ACCEPTED	0.06 s	details
#62	ACCEPTED	0.04 s	details
#63	ACCEPTED	0.04 s	details
#64	ACCEPTED	0.04 s	details
#65	ACCEPTED	0.04 s	details
#66	ACCEPTED	0.04 s	details
#67	ACCEPTED	0.06 s	details
#68	ACCEPTED	0.04 s	details
#69	ACCEPTED	0.04 s	details
#70	ACCEPTED	0.04 s	details
#71	ACCEPTED	0.04 s	details
#72	ACCEPTED	0.04 s	details
#73	ACCEPTED	0.06 s	details
#74	TIME LIMIT EXCEEDED	--	details
#75	TIME LIMIT EXCEEDED	--	details
#76	TIME LIMIT EXCEEDED	--	details
#77	TIME LIMIT EXCEEDED	--	details
#78	TIME LIMIT EXCEEDED	--	details
#79	TIME LIMIT EXCEEDED	--	details
#80	TIME LIMIT EXCEEDED	--	details
#81	TIME LIMIT EXCEEDED	--	details
#82	RUNTIME ERROR	0.11 s	details
#83	RUNTIME ERROR	0.11 s	details
#84	RUNTIME ERROR	0.11 s	details
#85	RUNTIME ERROR	0.11 s	details
#86	RUNTIME ERROR	0.12 s	details
#87	RUNTIME ERROR	0.14 s	details
#88	RUNTIME ERROR	0.14 s	details
#89	RUNTIME ERROR	0.14 s	details
#90	ACCEPTED	0.04 s	details
#91	ACCEPTED	0.04 s	details
#92	ACCEPTED	0.04 s	details
#93	ACCEPTED	0.04 s	details
#94	ACCEPTED	0.06 s	details
#95	TIME LIMIT EXCEEDED	--	details
#96	TIME LIMIT EXCEEDED	--	details
#97	RUNTIME ERROR	0.11 s	details
#98	RUNTIME ERROR	0.11 s	details
#99	RUNTIME ERROR	0.14 s	details
#100	ACCEPTED	0.04 s	details

Code

import sys

def main():
    input_data = sys.stdin.read().split()
    
    N = int(input_data[0])
    C = [int(x) for x in input_data[1:N+1]]
    
    values = [C, [-x for x in C], [0] * N] #black, red, green

    def get_max(idx, prev_color, start_color):
        if idx == N - 1:
            #color conflict
            choices = [values[c][idx] for c in range(3) if c != prev_color and c != start_color]
            return max(choices)
        
        #elimination
        choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
        return max(choices)

    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
    print(ans)

main()

Test details

Test 1

Verdict: ACCEPTED

input
`3 41702 99720 72033` view save

correct output
`58018` view save

user output
`58018` view save

Test 2

Verdict: ACCEPTED

input
`4 41702 99720 72033 93257` view save

correct output
`192977` view save

user output
`192977` view save

Test 3

Verdict: ACCEPTED

input
`5 41702 99720 72033 93257 11` view save

correct output
`192966` view save

user output
`192966` view save

Test 4

Verdict: ACCEPTED

input
`6 41702 99720 72033 93257 11 128...` view save

correct output
`205789` view save

user output
`205789` view save

Test 5

Verdict: ACCEPTED

input
`7 41702 99720 72033 93257 11 128...` view save

correct output
`223199` view save

user output
`223199` view save

Test 6

Verdict: ACCEPTED

input
`10 41702 99720 72033 93257 11 128...` view save

correct output
`329303` view save

user output
`329303` view save

Test 7

Verdict: ACCEPTED

input
`3 43600 18508 2592` view save

correct output
`41008` view save

user output
`41008` view save

Test 8

Verdict: ACCEPTED

input
`4 43600 18508 2592 93155` view save

correct output
`111663` view save

user output
`111663` view save

Test 9

Verdict: ACCEPTED

input
`5 43600 18508 2592 93155 54967` view save

correct output
`134163` view save

user output
`134163` view save

Test 10

Verdict: ACCEPTED

input
`6 43600 18508 2592 93155 54967 9...` view save

correct output
`206437` view save

user output
`206437` view save

Test 11

Verdict: ACCEPTED

input
`7 43600 18508 2592 93155 54967 9...` view save

correct output
`228937` view save

user output
`228937` view save

Test 12

Verdict: ACCEPTED

input
`10 43600 18508 2592 93155 54967 9...` view save

correct output
`286966` view save

user output
`286966` view save

Test 13

Verdict: ACCEPTED

input
`3 55080 7072 70815` view save

correct output
`63743` view save

user output
`63743` view save

Test 14

Verdict: ACCEPTED

input
`4 55080 7072 70815 83996` view save

correct output
`125895` view save

user output
`125895` view save

Test 15

Verdict: ACCEPTED

input
`5 55080 7072 70815 83996 29090` view save

correct output
`132004` view save

user output
`132004` view save

Test 16

Verdict: ACCEPTED

input
`6 55080 7072 70815 83996 29090 1...` view save

correct output
`154985` view save

user output
`154985` view save

Test 17

Verdict: ACCEPTED

input
`7 55080 7072 70815 83996 29090 1...` view save

correct output
`144137` view save

user output
`144137` view save

Test 18

Verdict: ACCEPTED

input
`10 96704 90063 54724 17269 97269 ...` view save

correct output
`389952` view save

user output
`389952` view save

Test 19

Verdict: ACCEPTED

input
`3 96704 90063 54724` view save

correct output
`41980` view save

user output
`41980` view save

Test 20

Verdict: ACCEPTED

input
`4 96704 90063 54724 17269` view save

correct output
`151428` view save

user output
`151428` view save

Test 21

Verdict: ACCEPTED

input
`5 96704 90063 54724 17269 97269` view save

correct output
`170063` view save

user output
`170063` view save

Test 22

Verdict: ACCEPTED

input
`6 96704 90063 54724 17269 97269 ...` view save

correct output
`248697` view save

user output
`248697` view save

Test 23

Verdict: ACCEPTED

input
`7 96704 90063 54724 17269 97269 ...` view save

correct output
`241545` view save

user output
`241545` view save

Test 24

Verdict: ACCEPTED

input
`10 96704 90063 54724 17269 97269 ...` view save

correct output
`389952` view save

user output
`389952` view save

Test 25

Verdict: TIME LIMIT EXCEEDED

input
`50 41702 99720 72033 93257 11 128...` view save

correct output
`1508857` view save

user output
(empty)

Test 26

Verdict: TIME LIMIT EXCEEDED

input
`50 43600 18508 2592 93155 54967 9...` view save

correct output
`1261022` view save

user output
(empty)

Test 27

Verdict: TIME LIMIT EXCEEDED

input
`50 55080 7072 70815 83996 29090 1...` view save

correct output
`1204485` view save

user output
(empty)

Test 28

Verdict: TIME LIMIT EXCEEDED

input
`50 96704 90063 54724 17269 97269 ...` view save

correct output
`1548881` view save

user output
(empty)

Test 29

Verdict: TIME LIMIT EXCEEDED

input
`250 41702 99720 72033 93257 11 128...` view save

correct output
`7048856` view save

user output
(empty)

Test 30

Verdict: TIME LIMIT EXCEEDED

input
`250 43600 18508 2592 93155 54967 9...` view save

correct output
`7355456` view save

user output
(empty)

Test 31

Verdict: TIME LIMIT EXCEEDED

input
`250 55080 7072 70815 83996 29090 1...` view save

correct output
`6879335` view save

user output
(empty)

Test 32

Verdict: TIME LIMIT EXCEEDED

input
`250 96704 90063 54724 17269 97269 ...` view save

correct output
`7241700` view save

user output
(empty)

Test 33

Verdict: RUNTIME ERROR

input
`900 41702 99720 72033 93257 11 128...` view save

correct output
`25585084` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 34

Verdict: RUNTIME ERROR

input
`900 43600 18508 2592 93155 54967 9...` view save

correct output
`25559325` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 35

Verdict: RUNTIME ERROR

input
`900 55080 7072 70815 83996 29090 1...` view save

correct output
`25395388` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 36

Verdict: RUNTIME ERROR

input
`900 96704 90063 54724 17269 97269 ...` view save

correct output
`25722421` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 37

Verdict: RUNTIME ERROR

input
`10000 41702 99720 72033 93257 11 128...` view save

correct output
`286456959` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 38

Verdict: ACCEPTED

input
`3 2 3 3` view save

correct output
`1` view save

user output
`1` view save

Test 39

Verdict: ACCEPTED

input
`4 2 4 3 4` view save

correct output
`8` view save

user output
`8` view save

Test 40

Verdict: ACCEPTED

input
`5 3 5 4 5 1` view save

correct output
`9` view save

user output
`9` view save

Test 41

Verdict: ACCEPTED

input
`6 3 6 5 6 1 1` view save

correct output
`13` view save

user output
`13` view save

Test 42

Verdict: ACCEPTED

input
`7 3 7 6 7 1 1 3` view save

correct output
`16` view save

user output
`16` view save

Test 43

Verdict: ACCEPTED

input
`10 5 10 8 10 1 2 4 10 2 3` view save

correct output
`35` view save

user output
`35` view save

Test 44

Verdict: ACCEPTED

input
`3 2 1 1` view save

correct output
`1` view save

user output
`1` view save

Test 45

Verdict: ACCEPTED

input
`4 2 1 1 4` view save

correct output
`5` view save

user output
`5` view save

Test 46

Verdict: ACCEPTED

input
`5 3 1 1 5 3` view save

correct output
`7` view save

user output
`7` view save

Test 47

Verdict: ACCEPTED

input
`6 3 2 1 6 4 6` view save

correct output
`14` view save

user output
`14` view save

Test 48

Verdict: ACCEPTED

input
`7 4 2 1 7 4 7 4` view save

correct output
`17` view save

user output
`17` view save

Test 49

Verdict: ACCEPTED

input
`10 5 2 1 10 6 10 5 5 5 4` view save

correct output
`31` view save

user output
`31` view save

Test 50

Verdict: ACCEPTED

input
`3 2 3 3` view save

correct output
`1` view save

user output
`1` view save

Test 51

Verdict: ACCEPTED

input
`4 2 4 3 4` view save

correct output
`8` view save

user output
`8` view save

Test 52

Verdict: ACCEPTED

input
`5 3 5 4 5 1` view save

correct output
`9` view save

user output
`9` view save

Test 53

Verdict: ACCEPTED

input
`6 3 6 5 6 1 1` view save

correct output
`13` view save

user output
`13` view save

Test 54

Verdict: ACCEPTED

input
`7 3 7 6 7 1 1 3` view save

correct output
`16` view save

user output
`16` view save

Test 55

Verdict: ACCEPTED

input
`10 5 10 8 10 1 2 4 10 2 3` view save

correct output
`35` view save

user output
`35` view save

Test 56

Verdict: ACCEPTED

input
`3 2 1 1` view save

correct output
`1` view save

user output
`1` view save

Test 57

Verdict: ACCEPTED

input
`4 2 1 1 4` view save

correct output
`5` view save

user output
`5` view save

Test 58

Verdict: ACCEPTED

input
`5 3 1 1 5 3` view save

correct output
`7` view save

user output
`7` view save

Test 59

Verdict: ACCEPTED

input
`6 3 2 1 6 4 6` view save

correct output
`14` view save

user output
`14` view save

Test 60

Verdict: ACCEPTED

input
`7 4 2 1 7 4 7 4` view save

correct output
`17` view save

user output
`17` view save

Test 61

Verdict: ACCEPTED

input
`10 5 2 1 10 6 10 5 5 5 4` view save

correct output
`31` view save

user output
`31` view save

Test 62

Verdict: ACCEPTED

input
`3 2 1 3` view save

correct output
`2` view save

user output
`2` view save

Test 63

Verdict: ACCEPTED

input
`4 3 1 3 4` view save

correct output
`6` view save

user output
`6` view save

Test 64

Verdict: ACCEPTED

input
`5 3 1 4 5 2` view save

correct output
`7` view save

user output
`7` view save

Test 65

Verdict: ACCEPTED

input
`6 4 1 5 6 2 1` view save

correct output
`11` view save

user output
`11` view save

Test 66

Verdict: ACCEPTED

input
`7 4 1 5 6 3 1 4` view save

correct output
`11` view save

user output
`11` view save

Test 67

Verdict: ACCEPTED

input
`10 10 10 6 2 10 9 8 7 7 6` view save

correct output
`41` view save

user output
`41` view save

Test 68

Verdict: ACCEPTED

input
`3 3 3 2` view save

correct output
`1` view save

user output
`1` view save

Test 69

Verdict: ACCEPTED

input
`4 4 4 3 1` view save

correct output
`7` view save

user output
`7` view save

Test 70

Verdict: ACCEPTED

input
`5 5 5 3 1 5` view save

correct output
`9` view save

user output
`9` view save

Test 71

Verdict: ACCEPTED

input
`6 6 6 4 2 6 6` view save

correct output
`16` view save

user output
`16` view save

Test 72

Verdict: ACCEPTED

input
`7 7 7 4 2 7 6 6` view save

correct output
`18` view save

user output
`18` view save

Test 73

Verdict: ACCEPTED

input
`10 10 10 6 2 10 9 8 7 7 6` view save

correct output
`41` view save

user output
`41` view save

Test 74

Verdict: TIME LIMIT EXCEEDED

input
`50 21 50 37 47 1 7 16 50 8 12 5 2...` view save

correct output
`764` view save

user output
(empty)

Test 75

Verdict: TIME LIMIT EXCEEDED

input
`50 22 10 2 47 28 48 22 25 22 17 1...` view save

correct output
`641` view save

user output
(empty)

Test 76

Verdict: TIME LIMIT EXCEEDED

input
`50 28 4 36 42 15 7 26 29 45 22 45...` view save

correct output
`610` view save

user output
(empty)

Test 77

Verdict: TIME LIMIT EXCEEDED

input
`50 49 46 28 9 49 43 36 31 35 30 1...` view save

correct output
`783` view save

user output
(empty)

Test 78

Verdict: TIME LIMIT EXCEEDED

input
`250 105 250 181 234 1 33 76 250 37...` view save

correct output
`17676` view save

user output
(empty)

Test 79

Verdict: TIME LIMIT EXCEEDED

input
`250 109 47 7 233 138 237 109 122 1...` view save

correct output
`18449` view save

user output
(empty)

Test 80

Verdict: TIME LIMIT EXCEEDED

input
`250 138 18 178 210 73 31 128 143 2...` view save

correct output
`17250` view save

user output
(empty)

Test 81

Verdict: TIME LIMIT EXCEEDED

input
`250 242 226 137 44 244 214 179 153...` view save

correct output
`18157` view save

user output
(empty)

Test 82

Verdict: RUNTIME ERROR

input
`900 376 898 649 840 1 116 273 900 ...` view save

correct output
`230450` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 83

Verdict: RUNTIME ERROR

input
`900 393 167 24 839 495 853 392 437...` view save

correct output
`230221` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 84

Verdict: RUNTIME ERROR

input
`900 496 64 638 756 262 110 460 513...` view save

correct output
`228745` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 85

Verdict: RUNTIME ERROR

input
`900 871 811 493 156 876 771 644 54...` view save

correct output
`231686` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 86

Verdict: RUNTIME ERROR

input
`10000 4171 9972 7204 9326 2 1282 302...` view save

correct output
`28647603` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 87

Verdict: RUNTIME ERROR

input
`100000 41703 99721 72034 93258 12 128...` view save

correct output
`2838931183` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 88

Verdict: RUNTIME ERROR

input
`100000 43601 18509 2593 93156 54968 9...` view save

correct output
`2842573127` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 89

Verdict: RUNTIME ERROR

input
`100000 55081 7073 70816 83997 29091 1...` view save

correct output
`2839271262` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 90

Verdict: ACCEPTED

input
`4 100000 100000 100000 100000` view save

correct output
`200000` view save

user output
`200000` view save

Test 91

Verdict: ACCEPTED

input
`5 100000 100000 100000 100000 10...` view save

correct output
`100000` view save

user output
`100000` view save

Test 92

Verdict: ACCEPTED

input
`6 100000 100000 100000 100000 10...` view save

correct output
`300000` view save

user output
`300000` view save

Test 93

Verdict: ACCEPTED

input
`7 100000 100000 100000 100000 10...` view save

correct output
`200000` view save

user output
`200000` view save

Test 94

Verdict: ACCEPTED

input
`10 100000 100000 100000 100000 10...` view save

correct output
`500000` view save

user output
`500000` view save

Test 95

Verdict: TIME LIMIT EXCEEDED

input
`50 100000 100000 100000 100000 10...` view save

correct output
`2500000` view save

user output
(empty)

Test 96

Verdict: TIME LIMIT EXCEEDED

input
`250 100000 100000 100000 100000 10...` view save

correct output
`12500000` view save

user output
(empty)

Test 97

Verdict: RUNTIME ERROR

input
`900 100000 100000 100000 100000 10...` view save

correct output
`45000000` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 98

Verdict: RUNTIME ERROR

input
`10000 100000 100000 100000 100000 10...` view save

correct output
`500000000` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 99

Verdict: RUNTIME ERROR

input
`100000 100000 100000 100000 100000 10...` view save

correct output
`5000000000` view save

user output
(empty)

Error:

Traceback (most recent call last):
  File "/box/input/code.py", line 24, in <module>
    main()
  File "/box/input/code.py", line 21, in main
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 21, in <genexpr>
    ans = max(values[start][0] + get_max(1, start, start) for start in range(3))
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in get_max
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", line 18, in <listcomp>
    choices = [values[c][idx] + get_max(idx + 1, c, start_color) for c in range(3) if c != prev_color]
  File "/box/input/code.py", li...

Test 100

Verdict: ACCEPTED

input
`3 100000 100000 100000` view save

correct output
`0` view save

user output
`0` view save

Aalto Competitive Programming 2026 - wk3 - Mon

Wheel of fortune

Code

Test details

Test 1

Test 2

Test 3

Test 4

Test 5

Test 6

Test 7

Test 8

Test 9

Test 10

Test 11

Test 12

Test 13

Test 14

Test 15

Test 16

Test 17

Test 18

Test 19

Test 20

Test 21

Test 22

Test 23

Test 24

Test 25

Test 26

Test 27

Test 28

Test 29

Test 30

Test 31

Test 32

Test 33

Test 34

Test 35

Test 36

Test 37

Test 38

Test 39

Test 40

Test 41

Test 42

Test 43

Test 44

Test 45

Test 46

Test 47

Test 48

Test 49

Test 50

Test 51

Test 52

Test 53

Test 54

Test 55

Test 56

Test 57

Test 58

Test 59

Test 60

Test 61

Test 62

Test 63

Test 64

Test 65

Test 66

Test 67

Test 68

Test 69

Test 70

Test 71

Test 72

Test 73

Test 74

Test 75

Test 76