CSES - Aalto Competitive Programming 2025 - wk3 - Mon

Submission details
Task:	Wheel of fortune
Sender:	Ranjana
Submission time:	2025-09-15 17:27:07 +0300
Language:	Python3 (PyPy3)
Status:	READY
Result:	WRONG ANSWER

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

Code

N = int(input())
C = list(map(int, input().split()))

def solve(C, first_color=None):
    N = len(C)
    dp = [[0]*3 for _ in range(N)]
    
    
    if first_color is None: 
        dp[0][0] = 0
        dp[0][1] = C[0]
        dp[0][2] = -C[0]
    else:
        dp[0][0] = 0
        dp[0][1] = C[0] if first_color == 1 else -float('inf')
        dp[0][2] = -C[0] if first_color == 2 else -float('inf')
    
    for i in range(1, N):
        # green can follow any
        dp[i][0] = max(dp[i-1])
        # black cannot follow black
        dp[i][1] = max(dp[i-1][0], dp[i-1][2]) + C[i]
        # red cannot follow red
        dp[i][2] = max(dp[i-1][0], dp[i-1][1]) - C[i]
    
    if first_color in [1,2]:
        
        result = max(dp[N-1][c] for c in range(3) if c != first_color)
    else:
        result = max(dp[N-1])
    return result


case1 = solve(C, first_color=None)

case2 = solve(C, first_color=1)

case3 = solve(C, first_color=2)

print(max(case1, case2, case3))

Test details

Test 1

Verdict: WRONG ANSWER

input
`3 41702 99720 72033` view save

correct output
`58018` view save

user output
`113735` 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: WRONG ANSWER

input
`5 41702 99720 72033 93257 11` view save

correct output
`192966` view save

user output
`192977` 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: WRONG ANSWER

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

correct output
`223199` view save

user output
`223210` 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: WRONG ANSWER

input
`3 43600 18508 2592` view save

correct output
`41008` view save

user output
`46192` view save

Test 8

Verdict: WRONG ANSWER

input
`4 43600 18508 2592 93155` view save

correct output
`111663` view save

user output
`136755` view save

Test 9

Verdict: WRONG ANSWER

input
`5 43600 18508 2592 93155 54967` view save

correct output
`134163` view save

user output
`136755` view save

Test 10

Verdict: WRONG ANSWER

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

correct output
`206437` view save

user output
`231529` view save

Test 11

Verdict: WRONG ANSWER

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

correct output
`228937` view save

user output
`231529` view save

Test 12

Verdict: WRONG ANSWER

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

correct output
`286966` view save

user output
`312058` view save

Test 13

Verdict: WRONG ANSWER

input
`3 55080 7072 70815` view save

correct output
`63743` view save

user output
`125895` view save

Test 14

Verdict: WRONG ANSWER

input
`4 55080 7072 70815 83996` view save

correct output
`125895` view save

user output
`139076` view save

Test 15

Verdict: WRONG ANSWER

input
`5 55080 7072 70815 83996 29090` view save

correct output
`132004` view save

user output
`154985` 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: WRONG ANSWER

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

correct output
`144137` view save

user output
`206068` 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: WRONG ANSWER

input
`3 96704 90063 54724` view save

correct output
`41980` view save

user output
`151428` 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: WRONG ANSWER

input
`5 96704 90063 54724 17269 97269` view save

correct output
`170063` view save

user output
`248697` 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: WRONG ANSWER

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

correct output
`241545` view save

user output
`320179` 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: WRONG ANSWER

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

correct output
`1508857` view save

user output
`1543827` view save

Test 26

Verdict: WRONG ANSWER

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

correct output
`1261022` view save

user output
`1314928` view save

Test 27

Verdict: WRONG ANSWER

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

correct output
`1204485` view save

user output
`1306065` view save

Test 28

Verdict: WRONG ANSWER

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

correct output
`1548881` view save

user output
`1609766` view save

Test 29

Verdict: WRONG ANSWER

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

correct output
`7048856` view save

user output
`7270303` view save

Test 30

Verdict: WRONG ANSWER

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

correct output
`7355456` view save

user output
`7593904` view save

Test 31

Verdict: WRONG ANSWER

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

correct output
`6879335` view save

user output
`7162979` view save

Test 32

Verdict: WRONG ANSWER

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

correct output
`7241700` view save

user output
`7419908` view save

Test 33

Verdict: WRONG ANSWER

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

correct output
`25585084` view save

user output
`26278635` view save

Test 34

Verdict: WRONG ANSWER

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

correct output
`25559325` view save

user output
`26412908` view save

Test 35

Verdict: WRONG ANSWER

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

correct output
`25395388` view save

user output
`26328113` view save

Test 36

Verdict: WRONG ANSWER

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

correct output
`25722421` view save

user output
`26484876` view save

Test 37

Verdict: WRONG ANSWER

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

correct output
`286456959` view save

user output
`295025669` view save

Test 38

Verdict: WRONG ANSWER

input
`3 2 3 3` view save

correct output
`1` view save

user output
`5` 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: WRONG ANSWER

input
`5 3 5 4 5 1` view save

correct output
`9` view save

user output
`10` 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: WRONG ANSWER

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

correct output
`16` view save

user output
`17` 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: WRONG ANSWER

input
`3 2 1 1` view save

correct output
`1` view save

user output
`3` view save

Test 45

Verdict: WRONG ANSWER

input
`4 2 1 1 4` view save

correct output
`5` view save

user output
`6` view save

Test 46

Verdict: WRONG ANSWER

input
`5 3 1 1 5 3` view save

correct output
`7` view save

user output
`8` view save

Test 47

Verdict: WRONG ANSWER

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

correct output
`14` view save

user output
`15` view save

Test 48

Verdict: WRONG ANSWER

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

correct output
`17` view save

user output
`18` view save

Test 49

Verdict: WRONG ANSWER

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

correct output
`31` view save

user output
`34` view save

Test 50

Verdict: WRONG ANSWER

input
`3 2 3 3` view save

correct output
`1` view save

user output
`5` 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: WRONG ANSWER

input
`5 3 5 4 5 1` view save

correct output
`9` view save

user output
`10` 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: WRONG ANSWER

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

correct output
`16` view save

user output
`17` 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: WRONG ANSWER

input
`3 2 1 1` view save

correct output
`1` view save

user output
`3` view save

Test 57

Verdict: WRONG ANSWER

input
`4 2 1 1 4` view save

correct output
`5` view save

user output
`6` view save

Test 58

Verdict: WRONG ANSWER

input
`5 3 1 1 5 3` view save

correct output
`7` view save

user output
`8` view save

Test 59

Verdict: WRONG ANSWER

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

correct output
`14` view save

user output
`15` view save

Test 60

Verdict: WRONG ANSWER

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

correct output
`17` view save

user output
`18` view save

Test 61

Verdict: WRONG ANSWER

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

correct output
`31` view save

user output
`34` view save

Test 62

Verdict: WRONG ANSWER

input
`3 2 1 3` view save

correct output
`2` view save

user output
`5` view save

Test 63

Verdict: WRONG ANSWER

input
`4 3 1 3 4` view save

correct output
`6` view save

user output
`7` view save

Test 64

Verdict: WRONG ANSWER

input
`5 3 1 4 5 2` view save

correct output
`7` view save

user output
`9` 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: WRONG ANSWER

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

correct output
`11` view save

user output
`16` 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: WRONG ANSWER

input
`3 3 3 2` view save

correct output
`1` view save

user output
`5` 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: WRONG ANSWER

input
`5 5 5 3 1 5` view save

correct output
`9` view save

user output
`13` 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: WRONG ANSWER

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

correct output
`18` view save

user output
`24` 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: WRONG ANSWER

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

correct output
`764` view save

user output
`780` view save

Test 75

Verdict: WRONG ANSWER

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

correct output
`641` view save

user output
`668` view save

Test 76

Verdict: WRONG ANSWER

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

correct output
`610` view save

user output
`663` view save

Test 77

Verdict: WRONG ANSWER

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

correct output
`783` view save

user output
`816` view save

Test 78

Verdict: WRONG ANSWER

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

correct output
`17676` view save

user output
`18231` view save

Test 79

Verdict: WRONG ANSWER

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

correct output
`18449` view save

user output
`19044` view save

Test 80

Verdict: WRONG ANSWER

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

correct output
`17250` view save

user output
`17965` view save

Test 81

Verdict: WRONG ANSWER

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

correct output
`18157` view save

user output
`18603` view save

Test 82

Verdict: WRONG ANSWER

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

correct output
`230450` view save

user output
`236705` view save

Test 83

Verdict: WRONG ANSWER

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

correct output
`230221` view save

user output
`237920` view save

Test 84

Verdict: WRONG ANSWER

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

correct output
`228745` view save

user output
`237161` view save

Test 85

Verdict: WRONG ANSWER

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

correct output
`231686` view save

user output
`238569` view save

Test 86

Verdict: WRONG ANSWER

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

correct output
`28647603` view save

user output
`29504688` view save

Test 87

Verdict: WRONG ANSWER

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

correct output
`2838931183` view save

user output
`2927730413` view save

Test 88

Verdict: WRONG ANSWER

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

correct output
`2842573127` view save

user output
`2932674995` view save

Test 89

Verdict: WRONG ANSWER

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

correct output
`2839271262` view save

user output
`2930651957` view save

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: WRONG ANSWER

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

correct output
`100000` view save

user output
`300000` 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: WRONG ANSWER

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

correct output
`200000` view save

user output
`400000` 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: ACCEPTED

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

correct output
`2500000` view save

user output
`2500000` view save

Test 96

Verdict: ACCEPTED

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

correct output
`12500000` view save

user output
`12500000` view save

Test 97

Verdict: ACCEPTED

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

correct output
`45000000` view save

user output
`45000000` view save

Test 98

Verdict: ACCEPTED

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

correct output
`500000000` view save

user output
`500000000` view save

Test 99

Verdict: ACCEPTED

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

correct output
`5000000000` view save

user output
`5000000000` view save

Test 100

Verdict: WRONG ANSWER

input
`3 100000 100000 100000` view save

correct output
`0` view save

user output
`200000` view save

Aalto Competitive Programming 2025 - 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