CSES - Aalto Competitive Programming 2025 - wk9 - Wed

Submission details
Task:	Good grades
Sender:	aalto25j_002
Submission time:	2025-11-05 21:14:49 +0200
Language:	Python3 (PyPy3)
Status:	READY
Result:	WRONG ANSWER

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.04 s	details
#7	ACCEPTED	0.04 s	details
#8	ACCEPTED	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	ACCEPTED	0.04 s	details
#14	ACCEPTED	0.04 s	details
#15	WRONG ANSWER	0.04 s	details
#16	ACCEPTED	0.04 s	details
#17	ACCEPTED	0.04 s	details
#18	ACCEPTED	0.04 s	details
#19	ACCEPTED	0.04 s	details
#20	WRONG ANSWER	0.04 s	details
#21	WRONG ANSWER	0.04 s	details
#22	WRONG ANSWER	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	ACCEPTED	0.04 s	details
#27	ACCEPTED	0.04 s	details
#28	WRONG ANSWER	0.04 s	details
#29	ACCEPTED	0.04 s	details
#30	WRONG ANSWER	0.06 s	details
#31	WRONG ANSWER	0.07 s	details
#32	WRONG ANSWER	0.07 s	details
#33	WRONG ANSWER	0.07 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.06 s	details
#38	WRONG ANSWER	0.06 s	details
#39	WRONG ANSWER	0.05 s	details
#40	WRONG ANSWER	0.08 s	details
#41	WRONG ANSWER	0.08 s	details
#42	WRONG ANSWER	0.09 s	details
#43	WRONG ANSWER	0.09 s	details
#44	WRONG ANSWER	0.08 s	details
#45	WRONG ANSWER	0.07 s	details
#46	WRONG ANSWER	0.09 s	details
#47	WRONG ANSWER	0.07 s	details
#48	WRONG ANSWER	0.08 s	details
#49	WRONG ANSWER	0.06 s	details
#50	WRONG ANSWER	0.30 s	details
#51	WRONG ANSWER	0.28 s	details
#52	WRONG ANSWER	0.39 s	details
#53	WRONG ANSWER	0.43 s	details
#54	WRONG ANSWER	0.33 s	details
#55	WRONG ANSWER	0.20 s	details
#56	WRONG ANSWER	0.33 s	details
#57	WRONG ANSWER	0.12 s	details
#58	WRONG ANSWER	0.33 s	details
#59	WRONG ANSWER	0.09 s	details

Code

def solve():

    n, k = map(int, input().split())
    s = list(map(int, input().split()))

    # dp[i][j] = max total smarts using first i students divided into j groups
    dp = [[-1] * (k + 1) for _ in range(n + 1)]
    dp[0][0] = 0

    # To reconstruct groups
    parent = [[-1] * (k + 1) for _ in range(n + 1)]

    for i in range(1, n + 1):
        for j in range(1, min(i, k) + 1):
            min_smart = s[i - 1]
            for l in range(i, j - 1, -1):  # l from i down to j
                min_smart = min(min_smart, s[l - 1])
                group_size = i - l + 1
                val = dp[l - 1][j - 1]
                if val == -1:
                    continue
                curr = val + group_size * min_smart
                if curr > dp[i][j]:
                    dp[i][j] = curr
                    parent[i][j] = l - 1

    # Reconstruct groups
    groups = [0] * n
    cur_i = n
    cur_j = k
    group_id = k
    while cur_j > 0:
        start = parent[cur_i][cur_j]
        for idx in range(start, cur_i):
            groups[idx] = group_id
        cur_i = start
        cur_j -= 1
        group_id -= 1

    print(" ".join(map(str, groups)))

solve()

Test details

Test 1

Verdict: ACCEPTED

input
`1 1 8` view save

correct output
`1` view save

user output
`1` view save

Test 2

Verdict: ACCEPTED

input
`2 2 7 6` view save

correct output
`2 1` view save

user output
`1 2` view save

Test 3

Verdict: ACCEPTED

input
`2 1 7 7` view save

correct output
`1 1` view save

user output
`1 1` view save

Test 4

Verdict: ACCEPTED

input
`2 2 7 1` view save

correct output
`2 1` view save

user output
`1 2` view save

Test 5

Verdict: ACCEPTED

input
`3 1 5 10 8` view save

correct output
`1 1 1` view save

user output
`1 1 1` view save

Test 6

Verdict: ACCEPTED

input
`3 1 6 10 2` view save

correct output
`1 1 1` view save

user output
`1 1 1` view save

Test 7

Verdict: ACCEPTED

input
`4 1 10 10 4 2` view save

correct output
`1 1 1 1` view save

user output
`1 1 1 1` view save

Test 8

Verdict: ACCEPTED

input
`4 4 5 1 5 4` view save

correct output
`3 1 4 2` view save

user output
`1 2 3 4` view save

Test 9

Verdict: WRONG ANSWER

input
`4 2 1 6 7 1` view save

correct output
`1 2 2 1` view save

user output
`1 1 1 2` view save

Test 10

Verdict: WRONG ANSWER

input
`5 3 6 8 9 7 9` view save

correct output
`1 2 3 1 3` view save

user output
`1 2 2 2 3` view save

Test 11

Verdict: WRONG ANSWER

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

correct output
`3 2 3 1 1` view save

user output
`1 1 2 3 3` view save

Test 12

Verdict: WRONG ANSWER

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

correct output
`1 1 3 2 3` view save

user output
`1 1 2 2 3` view save

Test 13

Verdict: ACCEPTED

input
`5 3 1 8 9 3 2` view save

correct output
`1 3 3 2 1` view save

user output
`1 2 2 3 3` view save

Test 14

Verdict: ACCEPTED

input
`5 5 10 6 2 10 9` view save

correct output
`4 2 1 5 3` view save

user output
`1 2 3 4 5` view save

Test 15

Verdict: WRONG ANSWER

input
`5 2 1 9 9 3 4` view save

correct output
`1 2 2 1 1` view save

user output
`1 2 2 2 2` view save

Test 16

Verdict: ACCEPTED

input
`5 5 10 4 3 9 1` view save

correct output
`5 3 2 4 1` view save

user output
`1 2 3 4 5` view save

Test 17

Verdict: ACCEPTED

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

correct output
`1 1 1 1 1` view save

user output
`1 1 1 1 1` view save

Test 18

Verdict: ACCEPTED

input
`5 5 1 10 3 9 4` view save

correct output
`1 5 2 4 3` view save

user output
`1 2 3 4 5` view save

Test 19

Verdict: ACCEPTED

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

correct output
`1 1 1 1 1` view save

user output
`1 1 1 1 1` view save

Test 20

Verdict: WRONG ANSWER

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

correct output
`2 4 5 3 5 2 6 1 3 3` view save

user output
`1 2 2 2 2 3 4 5 6 6` view save

Test 21

Verdict: WRONG ANSWER

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

correct output
`5 4 5 1 2 3 5 2 2 1` view save

user output
`1 1 1 2 2 3 4 5 5 5` view save

Test 22

Verdict: WRONG ANSWER

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

correct output
`1 1 5 4 5 3 3 3 2 2` view save

user output
`1 1 2 3 4 5 5 5 5 5` view save

Test 23

Verdict: WRONG ANSWER

input
`10 6 1 8 9 3 2 6 6 9 5 9` view save

correct output
`1 5 6 2 1 4 4 6 3 6` view save

user output
`1 2 2 3 3 4 4 4 5 6` view save

Test 24

Verdict: ACCEPTED

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

correct output
`9 3 1 10 8 7 5 6 4 2` view save

user output
`1 2 3 4 5 6 7 8 9 10` view save

Test 25

Verdict: WRONG ANSWER

input
`10 3 1 9 9 3 4 10 10 5 1 7` view save

correct output
`1 3 3 1 2 3 3 2 1 2` view save

user output
`1 1 1 1 1 2 2 3 3 3` view save

Test 26

Verdict: ACCEPTED

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

correct output
`8 4 3 7 1 1 5 2 9 6` view save

user output
`1 2 3 4 5 5 6 7 8 9` view save

Test 27

Verdict: ACCEPTED

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

correct output
`1 1 1 1 1 1 1 1 1 1` view save

user output
`1 1 1 1 1 1 1 1 1 1` view save

Test 28

Verdict: WRONG ANSWER

input
`10 9 1 10 3 9 4 6 9 3 5 1` view save

correct output
`1 9 2 7 4 6 8 3 5 1` view save

user output
`1 2 3 4 5 5 6 7 8 9` view save

Test 29

Verdict: ACCEPTED

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

correct output
`1 1 1 1 1 1 1 1 1 1` view save

user output
`1 1 1 1 1 1 1 1 1 1` view save

Test 30

Verdict: WRONG ANSWER

input
`100 55 636562060 767928734 906523441 ...` view save

correct output
`32 42 50 33 51 29 50 23 35 37 ...` view save

user output
`1 1 1 1 2 3 4 5 6 6 7 7 7 8 9 ...` view save

Test 31

Verdict: WRONG ANSWER

input
`100 42 773442532 122816 137572579 324...` view save

correct output
`34 1 8 16 9 13 6 20 10 19 18 3...` view save

user output
`1 2 3 3 3 3 3 3 3 4 4 4 4 4 5 ...` view save

Test 32

Verdict: WRONG ANSWER

input
`100 44 198730372 27838076 590195590 4...` view save

correct output
`9 1 28 21 23 20 15 16 7 10 36 ...` view save

user output
`1 1 2 2 2 2 3 3 4 4 5 5 6 7 7 ...` view save

Test 33

Verdict: WRONG ANSWER

input
`100 56 75940263 760367935 901888417 3...` view save

correct output
`6 44 51 19 8 33 36 54 29 54 1 ...` view save

user output
`1 2 2 3 3 4 4 5 6 7 8 8 8 9 9 ...` view save

Test 34

Verdict: WRONG ANSWER

input
`100 97 967034924 587586158 185430194 ...` view save

correct output
`94 59 19 88 79 67 77 64 27 15 ...` view save

user output
`1 2 3 4 5 6 7 8 9 10 11 12 13 ...` view save

Test 35

Verdict: WRONG ANSWER

input
`100 23 59249204 934941692 892631472 2...` view save

correct output
`2 22 21 6 10 23 12 3 15 11 19 ...` view save

user output
`1 2 2 3 3 3 3 3 4 4 4 4 4 4 5 ...` view save

Test 36

Verdict: WRONG ANSWER

input
`100 90 356460601 224848374 881788059 ...` view save

correct output
`23 14 83 4 1 28 8 54 7 48 56 3...` view save

user output
`1 2 3 4 4 5 6 7 8 9 10 11 12 1...` view save

Test 37

Verdict: WRONG ANSWER

input
`100 8 244103474 837431431 342493822 ...` view save

correct output
`2 7 3 5 7 5 3 6 3 5 1 1 5 3 1 ...` view save

user output
`1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 ...` view save

Test 38

Verdict: WRONG ANSWER

input
`100 88 11934038 257096283 933290530 4...` view save

correct output
`1 24 83 40 56 80 23 44 1 62 46...` view save

user output
`1 2 3 4 5 6 7 8 9 10 11 12 13 ...` view save

Test 39

Verdict: WRONG ANSWER

input
`100 2 391337048 538883744 535937150 ...` view save

correct output
`1 2 2 2 1 1 1 1 1 1 2 1 1 1 1 ...` view save

user output
`1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 ...` view save

Test 40

Verdict: WRONG ANSWER

input
`200 110 636562060 767928734 906523441 ...` view save

correct output
`69 86 99 70 101 61 100 47 74 7...` view save

user output
`1 1 1 1 1 1 2 3 4 4 5 5 5 6 7 ...` view save

Test 41

Verdict: WRONG ANSWER

input
`200 84 773442532 122816 137572579 324...` view save

correct output
`66 1 14 28 16 23 11 36 18 35 3...` view save

user output
`1 2 3 3 3 3 3 3 3 4 4 4 4 4 4 ...` view save

Test 42

Verdict: WRONG ANSWER

input
`200 88 198730372 27838076 590195590 4...` view save

correct output
`17 3 53 42 45 40 29 30 13 18 6...` view save

user output
`1 1 2 2 2 2 3 3 4 4 5 5 6 7 7 ...` view save

Test 43

Verdict: WRONG ANSWER

input
`200 111 75940263 760367935 901888417 3...` view save

correct output
`8 85 100 35 14 62 71 106 51 10...` view save

user output
`1 2 2 3 3 4 4 5 6 7 8 8 8 9 9 ...` view save

Test 44

Verdict: WRONG ANSWER

input
`200 194 967034924 587586158 185430194 ...` view save

correct output
`185 114 39 173 151 128 147 124...` view save

user output
`1 2 3 4 5 6 7 8 9 10 11 12 13 ...` view save

Test 45

Verdict: WRONG ANSWER

input
`200 45 59249204 934941692 892631472 2...` view save

correct output
`3 43 41 11 19 45 25 5 30 21 37...` view save

user output
`1 2 2 3 3 3 3 3 4 4 4 4 4 4 5 ...` view save

Test 46

Verdict: WRONG ANSWER

input
`200 179 356460601 224848374 881788059 ...` view save

correct output
`54 33 162 8 4 61 15 113 14 99 ...` view save

user output
`1 2 3 4 4 5 6 7 8 9 10 11 12 1...` view save

Test 47

Verdict: WRONG ANSWER

input
`200 16 244103474 837431431 342493822 ...` view save

correct output
`4 14 6 8 13 9 6 10 5 10 2 2 8 ...` view save

user output
`1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 ...` view save

Test 48

Verdict: WRONG ANSWER

input
`200 175 11934038 257096283 933290530 4...` view save

correct output
`2 40 165 67 105 157 37 77 2 11...` view save

user output
`1 2 3 4 5 6 7 8 9 10 11 12 13 ...` view save

Test 49

Verdict: WRONG ANSWER

input
`200 3 391337048 538883744 535937150 ...` view save

correct output
`2 2 2 2 1 1 1 1 1 1 3 2 1 1 1 ...` view save

user output
`1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` view save

Test 50

Verdict: WRONG ANSWER

input
`500 275 636562060 767928734 906523441 ...` view save

correct output
`176 215 251 178 254 161 252 12...` view save

user output
`1 1 1 1 2 3 4 5 6 6 7 7 7 8 9 ...` view save

Test 51

Verdict: WRONG ANSWER

input
`500 209 773442532 122816 137572579 324...` view save

correct output
`163 1 31 72 37 57 23 89 45 87 ...` view save

user output
`1 2 3 3 3 3 3 3 3 4 4 4 4 4 4 ...` view save

Test 52

Verdict: WRONG ANSWER

input
`500 218 198730372 27838076 590195590 4...` view save

correct output
`42 8 130 102 113 97 73 75 34 4...` view save

user output
`1 1 2 2 2 2 3 3 3 3 4 4 5 6 6 ...` view save

Test 53

Verdict: WRONG ANSWER

input
`500 276 75940263 760367935 901888417 3...` view save

correct output
`20 205 249 85 32 149 165 267 1...` view save

user output
`1 2 2 3 3 4 4 5 6 7 8 8 8 8 8 ...` view save

Test 54

Verdict: WRONG ANSWER

input
`500 484 967034924 587586158 185430194 ...` view save

correct output
`469 290 97 438 372 322 367 317...` view save

user output
`1 2 3 4 5 6 7 8 9 10 11 12 13 ...` view save

Test 55

Verdict: WRONG ANSWER

input
`500 111 59249204 934941692 892631472 2...` view save

correct output
`5 104 100 24 42 109 55 9 70 46...` view save

user output
`1 2 2 3 3 3 3 3 4 4 4 4 4 4 5 ...` view save

Test 56

Verdict: WRONG ANSWER

input
`500 447 356460601 224848374 881788059 ...` view save

correct output
`154 88 394 30 18 168 46 279 43...` view save

user output
`1 2 3 4 4 5 6 7 8 9 10 11 12 1...` view save

Test 57

Verdict: WRONG ANSWER

input
`500 39 244103474 837431431 342493822 ...` view save

correct output
`9 33 13 18 31 19 13 23 11 21 3...` view save

user output
`1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 ...` view save

Test 58

Verdict: WRONG ANSWER

input
`500 437 11934038 257096283 933290530 4...` view save

correct output
`7 101 407 167 251 388 97 191 7...` view save

user output
`1 2 3 4 5 6 7 8 9 10 11 12 13 ...` view save

Test 59

Verdict: WRONG ANSWER

input
`500 6 391337048 538883744 535937150 ...` view save

correct output
`3 4 4 4 1 2 3 2 1 2 6 3 2 2 2 ...` view save

user output
`1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` view save

Aalto Competitive Programming 2025 - wk9 - Wed

Good grades

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

Tasks

Submissions (aalto25j_002)