CSES - Aalto Competitive Programming 2025 - wk9 - Wed

Submission details
Task:	Good grades
Sender:	ind1f
Submission time:	2025-11-05 17:01:14 +0200
Language:	C++ (C++17)
Status:	READY
Result:	ACCEPTED

Test results
test	verdict	time
#1	ACCEPTED	0.01 s	details
#2	ACCEPTED	0.01 s	details
#3	ACCEPTED	0.01 s	details
#4	ACCEPTED	0.01 s	details
#5	ACCEPTED	0.01 s	details
#6	ACCEPTED	0.01 s	details
#7	ACCEPTED	0.01 s	details
#8	ACCEPTED	0.01 s	details
#9	ACCEPTED	0.01 s	details
#10	ACCEPTED	0.01 s	details
#11	ACCEPTED	0.01 s	details
#12	ACCEPTED	0.01 s	details
#13	ACCEPTED	0.01 s	details
#14	ACCEPTED	0.01 s	details
#15	ACCEPTED	0.01 s	details
#16	ACCEPTED	0.01 s	details
#17	ACCEPTED	0.01 s	details
#18	ACCEPTED	0.01 s	details
#19	ACCEPTED	0.01 s	details
#20	ACCEPTED	0.01 s	details
#21	ACCEPTED	0.01 s	details
#22	ACCEPTED	0.01 s	details
#23	ACCEPTED	0.01 s	details
#24	ACCEPTED	0.01 s	details
#25	ACCEPTED	0.01 s	details
#26	ACCEPTED	0.01 s	details
#27	ACCEPTED	0.01 s	details
#28	ACCEPTED	0.01 s	details
#29	ACCEPTED	0.01 s	details
#30	ACCEPTED	0.01 s	details
#31	ACCEPTED	0.01 s	details
#32	ACCEPTED	0.01 s	details
#33	ACCEPTED	0.01 s	details
#34	ACCEPTED	0.01 s	details
#35	ACCEPTED	0.01 s	details
#36	ACCEPTED	0.01 s	details
#37	ACCEPTED	0.01 s	details
#38	ACCEPTED	0.01 s	details
#39	ACCEPTED	0.01 s	details
#40	ACCEPTED	0.01 s	details
#41	ACCEPTED	0.01 s	details
#42	ACCEPTED	0.01 s	details
#43	ACCEPTED	0.01 s	details
#44	ACCEPTED	0.01 s	details
#45	ACCEPTED	0.01 s	details
#46	ACCEPTED	0.01 s	details
#47	ACCEPTED	0.01 s	details
#48	ACCEPTED	0.01 s	details
#49	ACCEPTED	0.01 s	details
#50	ACCEPTED	0.06 s	details
#51	ACCEPTED	0.04 s	details
#52	ACCEPTED	0.04 s	details
#53	ACCEPTED	0.06 s	details
#54	ACCEPTED	0.09 s	details
#55	ACCEPTED	0.03 s	details
#56	ACCEPTED	0.09 s	details
#57	ACCEPTED	0.01 s	details
#58	ACCEPTED	0.08 s	details
#59	ACCEPTED	0.01 s	details

Code

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 5e2 + 5;

int n, k;
pair<int, int> s[N];
int g[N];
long long f[N][N];
int q[N][N];

long long cost(int l, int r) {
  return 1LL * s[l].first * (r - l + 1);
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  cin >> n >> k;
  for (int i = 1; i <= n; i++) {
    cin >> s[i].first;
    s[i].second = i;
  }
  sort(s + 1, s + n + 1);
  memset(f, -0x3f, sizeof(f));
  f[0][0] = 0;
  for (int i = 1; i <= n; i++) {
    for (int j = 1; j <= k; j++) {
      for (int p = i; p >= 1; p--) {
        if (cost(p, i) + f[p - 1][j - 1] > f[i][j]) {
          f[i][j] = cost(p, i) + f[p - 1][j - 1];
          q[i][j] = p;
        }
      }
    }
  }
  int x = n, y = k;
  int ptr = 1;
  while (y > 0) {
    int l = q[x][y];
    for (int z = l; z <= x; z++) {
      g[s[z].second] = ptr; 
    }
    x = l - 1;
    y--;
    ptr++;
  }
  for (int i = 1; i <= n; i++) {
    cout << g[i] << ' ';
  }
  return 0;
}

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
`2 4 1 3` view save

Test 9

Verdict: ACCEPTED

input
`4 2 1 6 7 1` view save

correct output
`1 2 2 1` view save

user output
`2 1 1 2` view save

Test 10

Verdict: ACCEPTED

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

correct output
`1 2 3 1 3` view save

user output
`3 2 1 3 1` view save

Test 11

Verdict: ACCEPTED

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

correct output
`3 2 3 1 1` view save

user output
`1 2 1 3 3` view save

Test 12

Verdict: ACCEPTED

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

correct output
`1 1 3 2 3` view save

user output
`3 3 1 2 1` 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
`3 1 1 2 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
`2 4 5 1 3` view save

Test 15

Verdict: ACCEPTED

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

correct output
`1 2 2 1 1` view save

user output
`2 1 1 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 3 4 2 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
`5 1 4 2 3` 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: ACCEPTED

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
`5 3 2 4 2 5 1 6 4 4` view save

Test 21

Verdict: ACCEPTED

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 2 1 5 4 3 1 4 4 5` view save

Test 22

Verdict: ACCEPTED

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
`5 5 1 2 1 3 3 3 4 4` view save

Test 23

Verdict: ACCEPTED

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
`6 2 1 5 6 3 3 1 4 1` 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
`2 8 10 1 3 4 6 5 7 9` view save

Test 25

Verdict: ACCEPTED

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
`3 1 1 3 2 1 1 2 3 2` 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
`2 6 7 3 9 9 5 8 1 4` 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: ACCEPTED

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
`9 1 8 3 6 4 2 7 5 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: ACCEPTED

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

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

user output
`24 14 6 23 5 27 6 33 21 19 35 ...` view save

Test 31

Verdict: ACCEPTED

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
`9 42 35 27 34 30 37 23 33 24 2...` view save

Test 32

Verdict: ACCEPTED

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
`36 44 17 24 22 25 30 29 38 35 ...` view save

Test 33

Verdict: ACCEPTED

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
`51 13 6 38 49 24 21 3 28 3 56 ...` view save

Test 34

Verdict: ACCEPTED

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

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

user output
`4 39 79 10 19 31 21 34 71 83 6...` view save

Test 35

Verdict: ACCEPTED

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
`22 2 3 18 14 1 12 21 9 13 5 14...` view save

Test 36

Verdict: ACCEPTED

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
`68 77 8 87 90 63 83 37 84 43 3...` view save

Test 37

Verdict: ACCEPTED

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
`7 2 6 4 2 4 6 3 6 4 8 8 4 6 8 ...` view save

Test 38

Verdict: ACCEPTED

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
`88 65 6 49 33 9 66 45 88 27 43...` view save

Test 39

Verdict: ACCEPTED

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
`2 1 1 1 2 2 2 2 2 2 1 2 2 2 2 ...` view save

Test 40

Verdict: ACCEPTED

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

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

user output
`42 25 12 41 10 50 11 64 37 34 ...` view save

Test 41

Verdict: ACCEPTED

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
`19 84 71 57 69 62 74 49 67 50 ...` view save

Test 42

Verdict: ACCEPTED

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
`72 86 36 47 44 49 60 59 76 71 ...` view save

Test 43

Verdict: ACCEPTED

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
`104 27 12 77 98 50 41 6 61 5 1...` view save

Test 44

Verdict: ACCEPTED

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

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

user output
`10 81 156 22 44 67 48 71 144 1...` view save

Test 45

Verdict: ACCEPTED

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
`43 3 5 35 27 1 21 41 16 25 9 2...` view save

Test 46

Verdict: ACCEPTED

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

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

user output
`126 147 18 172 176 119 165 67 ...` view save

Test 47

Verdict: ACCEPTED

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
`13 3 11 9 4 8 11 7 12 7 15 15 ...` view save

Test 48

Verdict: ACCEPTED

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
`174 136 11 109 71 19 139 99 17...` view save

Test 49

Verdict: ACCEPTED

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
`2 2 2 2 3 3 3 3 3 3 1 2 3 3 3 ...` view save

Test 50

Verdict: ACCEPTED

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

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

user output
`100 61 25 98 22 115 24 154 89 ...` view save

Test 51

Verdict: ACCEPTED

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
`47 209 179 138 173 153 187 121...` view save

Test 52

Verdict: ACCEPTED

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
`177 211 89 117 106 122 146 144...` view save

Test 53

Verdict: ACCEPTED

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

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

user output
`257 72 28 192 245 128 112 10 1...` view save

Test 54

Verdict: ACCEPTED

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

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

user output
`16 195 388 47 113 163 118 168 ...` view save

Test 55

Verdict: ACCEPTED

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
`107 8 12 88 70 3 57 103 42 66 ...` view save

Test 56

Verdict: ACCEPTED

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

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

user output
`294 360 54 418 430 280 402 169...` view save

Test 57

Verdict: ACCEPTED

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
`31 7 27 22 9 21 27 17 29 19 37...` view save

Test 58

Verdict: ACCEPTED

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

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

user output
`431 337 31 271 187 50 341 247 ...` view save

Test 59

Verdict: ACCEPTED

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
`4 3 3 3 6 5 4 5 6 5 1 4 5 5 5 ...` 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 (ind1f)