CSES - Aalto Competitive Programming 2025 - wk9 - Wed

Submission details
Task:	Good grades
Sender:	aalto25j_002
Submission time:	2025-11-05 17:30:37 +0200
Language:	C++ (C++17)
Status:	READY
Result:	WRONG ANSWER

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

Code

#include <stdint.h>
#include <stdlib.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdbool.h>

#define LOOP(i,s,e) for (int64_t i = (s); i < (e); i++)
#define SCAN(...) if (scanf(__VA_ARGS__) == 0) return EXIT_FAILURE

typedef struct p{
    int64_t v;
    int64_t i;
} p;

int comp(const void *a, const void *b){
    p ca = *((p*)a);
    p cb = *((p*)b);
    if (ca.v == cb.v) return 0;
    if (ca.v < cb.v) return -1;
    return 1;
}

int main() {
    int64_t n, k;
    SCAN("%ld %ld", &n, &k);

    p *sorted = (p*) malloc(2 * n * sizeof(p));

    LOOP(i, 0, n){
        SCAN("%ld", &sorted[i].v);
        sorted[i].i=i;
    }


    qsort(sorted, n, sizeof(p), comp);

    int64_t *groups = (int64_t *) malloc((k+1) * sizeof(int64_t));

    // group i goes from groups[i] to groups[i+1]

    LOOP(i, 1, k)
        groups[i] = i;
    groups[0] = 0;
    groups[k] = n;

    int64_t old_smart = 0;
    int64_t smart = 0;
    LOOP(i, 0, k)
        smart += sorted[groups[i]].v * (groups[i+1]-groups[i]);

    while (old_smart != smart){
        old_smart = smart;
        // increase right
        for(int64_t i = k-1; i > 0; i--){
            int64_t diff =
                - sorted[groups[i]].v * (groups[i+1]-groups[i])
                + sorted[groups[i]].v * (groups[i+1]-groups[i] + 1)
                - sorted[groups[i+1]].v * (groups[i+2]-groups[i+1])
                + sorted[groups[i+1]+1].v * (groups[i+2]-groups[i+1] - 1);
            while(diff > 0 && groups[i+1] < groups[i+2]){
                groups[i+1]++;
                diff =
                - sorted[groups[i]].v * (groups[i+1]-groups[i])
                + sorted[groups[i]].v * (groups[i+1]-groups[i] + 1)
                - sorted[groups[i+1]].v * (groups[i+2]-groups[i+1])
                + sorted[groups[i+1]+1].v * (groups[i+2]-groups[i+1] - 1);

            }
        }
        // decrease left
        LOOP(i, 1, k){
            int64_t diff =
                - sorted[groups[i]].v * (groups[i+1]-groups[i])
                + sorted[groups[i]-1].v * (groups[i+1]-groups[i] + 1)
                - sorted[groups[i-1]].v * (groups[i]-groups[i-1])
                + sorted[groups[i-1]].v * (groups[i]-groups[i-1] - 1);
            while(diff > 0 && groups[i] < groups[i+1]){
                groups[i]--;
                diff =
                - sorted[groups[i]].v * (groups[i+1]-groups[i])
                + sorted[groups[i]-1].v * (groups[i+1]-groups[i] + 1)
                - sorted[groups[i-1]].v * (groups[i]-groups[i-1])
                + sorted[groups[i-1]].v * (groups[i]-groups[i-1] - 1);
            }
        }
        // get number
        smart = 0;
        LOOP(i, 0, k)
            smart += sorted[groups[i]].v * (groups[i+1]-groups[i]);
    }

    int64_t *places = (int64_t*) malloc(n * sizeof(int64_t));

    LOOP(g, 0, k){
        LOOP(i, groups[g], groups[g+1])
            places[sorted[i].i] = g+1;
    }

    LOOP(i, 0, n)
        printf("%ld ", places[i]);
    printf("\n");




    return EXIT_SUCCESS;
}

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

input
`2 2 7 6` view save

correct output
`2 1` view save

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

input
`2 2 7 1` view save

correct output
`2 1` view save

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

input
`4 4 5 1 5 4` view save

correct output
`3 1 4 2` view save

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

Test 24

Verdict: WRONG ANSWER

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

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

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

user output
(empty)

Test 31

Verdict: RUNTIME ERROR

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
(empty)

Test 32

Verdict: RUNTIME ERROR

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
(empty)

Test 33

Verdict: RUNTIME ERROR

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
(empty)

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
`97 97 19 97 81 97 79 97 27 97 ...` Truncated

Test 35

Verdict: RUNTIME ERROR

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
(empty)

Test 36

Verdict: RUNTIME ERROR

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
(empty)

Test 37

Verdict: RUNTIME ERROR

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
(empty)

Test 38

Verdict: RUNTIME ERROR

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
(empty)

Test 39

Verdict: RUNTIME ERROR

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
(empty)

Test 40

Verdict: RUNTIME ERROR

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

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

user output
(empty)

Test 41

Verdict: RUNTIME ERROR

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
(empty)

Test 42

Verdict: RUNTIME ERROR

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
(empty)

Test 43

Verdict: RUNTIME ERROR

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
(empty)

Test 44

Verdict: RUNTIME ERROR

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

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

user output
(empty)

Test 45

Verdict: RUNTIME ERROR

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
(empty)

Test 46

Verdict: RUNTIME ERROR

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

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

user output
(empty)

Test 47

Verdict: RUNTIME ERROR

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
(empty)

Test 48

Verdict: RUNTIME ERROR

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
(empty)

Test 49

Verdict: RUNTIME ERROR

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
(empty)

Test 50

Verdict: RUNTIME ERROR

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

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

user output
(empty)

Test 51

Verdict: RUNTIME ERROR

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
(empty)

Test 52

Verdict: RUNTIME ERROR

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
(empty)

Test 53

Verdict: RUNTIME ERROR

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

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

user output
(empty)

Test 54

Verdict: RUNTIME ERROR

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

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

user output
(empty)

Test 55

Verdict: RUNTIME ERROR

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
(empty)

Test 56

Verdict: RUNTIME ERROR

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

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

user output
(empty)

Test 57

Verdict: RUNTIME ERROR

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
(empty)

Test 58

Verdict: RUNTIME ERROR

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

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

user output
(empty)

Test 59

Verdict: RUNTIME ERROR

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
(empty)

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)