CSES - Aalto Competitive Programming 2024 - wk9 - Wed - Results
Submission details
Task:Good grades
Sender:aalto2024j_006
Submission time:2024-11-06 17:25:42 +0200
Language:C++ (C++17)
Status:READY
Result:
Test results
testverdicttime
#1ACCEPTED0.00 sdetails
#2ACCEPTED0.00 sdetails
#3ACCEPTED0.00 sdetails
#4ACCEPTED0.00 sdetails
#5ACCEPTED0.00 sdetails
#6ACCEPTED0.00 sdetails
#7ACCEPTED0.00 sdetails
#8ACCEPTED0.00 sdetails
#9ACCEPTED0.00 sdetails
#10ACCEPTED0.00 sdetails
#11ACCEPTED0.00 sdetails
#12ACCEPTED0.00 sdetails
#13ACCEPTED0.00 sdetails
#14ACCEPTED0.00 sdetails
#15ACCEPTED0.00 sdetails
#16ACCEPTED0.00 sdetails
#17ACCEPTED0.00 sdetails
#18ACCEPTED0.00 sdetails
#19ACCEPTED0.00 sdetails
#20ACCEPTED0.00 sdetails
#21ACCEPTED0.00 sdetails
#22ACCEPTED0.00 sdetails
#23ACCEPTED0.00 sdetails
#24ACCEPTED0.00 sdetails
#250.00 sdetails
#26ACCEPTED0.00 sdetails
#27ACCEPTED0.00 sdetails
#28ACCEPTED0.00 sdetails
#29ACCEPTED0.00 sdetails
#300.00 sdetails
#310.00 sdetails
#320.00 sdetails
#330.00 sdetails
#34ACCEPTED0.00 sdetails
#350.00 sdetails
#360.00 sdetails
#370.00 sdetails
#38ACCEPTED0.00 sdetails
#390.00 sdetails
#400.00 sdetails
#410.00 sdetails
#420.00 sdetails
#430.00 sdetails
#44ACCEPTED0.01 sdetails
#450.00 sdetails
#46ACCEPTED0.01 sdetails
#470.00 sdetails
#480.00 sdetails
#490.00 sdetails
#500.01 sdetails
#510.01 sdetails
#520.01 sdetails
#530.01 sdetails
#540.01 sdetails
#550.01 sdetails
#560.01 sdetails
#570.00 sdetails
#580.01 sdetails
#590.00 sdetails

Compiler report

input/code.cpp: In function 'void solve()':
input/code.cpp:26:23: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   26 |     for (int i = 0; i < nums.size(); i++)
      |                     ~~^~~~~~~~~~~~~
input/code.cpp:39:23: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |     for (int j = 1; j < dp[0].size(); j++)
      |                     ~~^~~~~~~~~~~~~~
input/code.cpp:41:27: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::vector<std::pair<long long int, long long int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   41 |         for (int i = 1; i < dp.size(); i++)
      |                         ~~^~~~~~...

Code

#include <iostream>
#include <vector>
#include <array>
#include <string>
#include <algorithm>
#include <numeric>
#include <unordered_map>
#include <unordered_set>
#include <queue>
#include <climits>
#include <cmath>
#include <functional>
#include <type_traits>
#include <fstream>
#include <bitset>
 
#define int long long
using namespace std;
 

void solve() {
    int n, k;
    cin >> n >> k;
    
    vector<pair<int, int>> nums(n);
    for (int i = 0; i < nums.size(); i++)
    {
        int a;
        cin >> a;
        nums[i] = {a, i};
    }

    sort(nums.begin(), nums.end());

    vector<vector<pair<int, int>>> dp(nums.size()+1, vector<pair<int, int>>(k+1, {LLONG_MIN / 2, 0}));
    vector<vector<int>> dpMem(nums.size()+1, vector<int>(k+1, 0));
    dp[0][0] = {0, 0};

    for (int j = 1; j < dp[0].size(); j++)
    {
        for (int i = 1; i < dp.size(); i++)
        {
            // p1: create new, p2: continue other
            int p1 = dp[i-1][j-1].first + nums[i-1].first;
            int p2 = dp[i-1][j].first + dp[i-1][j].second;
            if (p1 >= p2)
            {
                dp[i][j] = {p1, nums[i-1].first};
                dpMem[i][j] = 1;
            }
            else
            {
                dp[i][j] = {p2, dp[i-1][j].second};
            }
            //cout << dpMem[i][j] << "-" << " ";
        }
        //cout << endl;
    }

    vector<int> res(n);

    int j = k;
    for (int i = dp.size()-1; i > 0; i--)
    {
        res[nums[i-1].second] = j;
        if (dpMem[i][j] == 1)
            j--;
    }

    for (int i = 0; i < res.size(); i++)
        cout << res[i] << " ";

}
 
 
signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
 
    int t = 1;
 
    //cin >> t;
    //ifstream f("testcase.txt");
    //cin.rdbuf(f.rdbuf());
 
    for (int i = 0; i < t; i++)
        solve();
 
    return 0;
}

Test details

Test 1

Verdict: ACCEPTED

input
1 1

correct output

user output

Test 2

Verdict: ACCEPTED

input
2 2
7 6 

correct output
2 1 

user output
2 1 

Test 3

Verdict: ACCEPTED

input
2 1
7 7 

correct output
1 1 

user output
1 1 

Test 4

Verdict: ACCEPTED

input
2 2
7 1 

correct output
2 1 

user output
2 1 

Test 5

Verdict: ACCEPTED

input
3 1
5 10 8 

correct output
1 1 1 

user output
1 1 1 

Test 6

Verdict: ACCEPTED

input
3 1
6 10 2 

correct output
1 1 1 

user output
1 1 1 

Test 7

Verdict: ACCEPTED

input
4 1
10 10 4 2 

correct output
1 1 1 1 

user output
1 1 1 1 

Test 8

Verdict: ACCEPTED

input
4 4
5 1 5 4 

correct output
3 1 4 2 

user output
3 1 4 2 

Test 9

Verdict: ACCEPTED

input
4 2
1 6 7 1 

correct output
1 2 2 1 

user output
1 2 2 1 

Test 10

Verdict: ACCEPTED

input
5 3
6 8 9 7 9 

correct output
1 2 3 1 3 

user output
1 2 3 1 3 

Test 11

Verdict: ACCEPTED

input
5 3
10 8 10 1 2 

correct output
3 2 3 1 1 

user output
3 2 3 1 1 

Test 12

Verdict: ACCEPTED

input
5 3
2 1 10 6 10 

correct output
1 1 3 2 3 

user output
1 1 3 2 3 

Test 13

Verdict: ACCEPTED

input
5 3
1 8 9 3 2 

correct output
1 3 3 2 1 

user output
1 3 3 2 1 

Test 14

Verdict: ACCEPTED

input
5 5
10 6 2 10 9 

correct output
4 2 1 5 3 

user output
4 2 1 5 3 

Test 15

Verdict: ACCEPTED

input
5 2
1 9 9 3 4 

correct output
1 2 2 1 1 

user output
1 2 2 1 1 

Test 16

Verdict: ACCEPTED

input
5 5
10 4 3 9 1 

correct output
5 3 2 4 1 

user output
5 3 2 4 1 

Test 17

Verdict: ACCEPTED

input
5 1
3 8 4 5 10 

correct output
1 1 1 1 1 

user output
1 1 1 1 1 

Test 18

Verdict: ACCEPTED

input
5 5
1 10 3 9 4 

correct output
1 5 2 4 3 

user output
1 5 2 4 3 

Test 19

Verdict: ACCEPTED

input
5 1
4 6 5 5 1 

correct output
1 1 1 1 1 

user output
1 1 1 1 1 

Test 20

Verdict: ACCEPTED

input
10 6
6 8 9 7 9 6 9 5 7 7 

correct output
2 4 5 3 5 2 6 1 3 3 

user output
2 4 5 3 5 2 6 1 3 3 

Test 21

Verdict: ACCEPTED

input
10 5
10 8 10 1 2 4 10 2 3 1 

correct output
5 4 5 1 2 3 5 2 2 1 

user output
5 4 5 1 2 3 5 2 2 1 

Test 22

Verdict: ACCEPTED

input
10 5
2 1 10 6 10 5 5 5 4 4 

correct output
1 1 5 4 5 3 3 3 2 2 

user output
1 1 5 4 5 3 3 3 2 2 

Test 23

Verdict: ACCEPTED

input
10 6
1 8 9 3 2 6 6 9 5 9 

correct output
1 5 6 2 1 4 4 6 3 6 

user output
1 5 6 2 1 4 4 6 3 6 

Test 24

Verdict: ACCEPTED

input
10 10
10 6 2 10 9 8 7 7 6 3 

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

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

Test 25

Verdict:

input
10 3
1 9 9 3 4 10 10 5 1 7 

correct output
1 3 3 1 2 3 3 2 1 2 

user output
1 3 3 2 2 3 3 2 1 3 

Test 26

Verdict: ACCEPTED

input
10 9
10 4 3 9 1 1 4 2 10 6 

correct output
8 4 3 7 1 1 5 2 9 6 

user output
8 4 3 7 1 1 5 2 9 6 

Test 27

Verdict: ACCEPTED

input
10 1
3 8 4 5 10 8 5 10 4 6 

correct output
1 1 1 1 1 1 1 1 1 1 

user output
1 1 1 1 1 1 1 1 1 1 

Test 28

Verdict: ACCEPTED

input
10 9
1 10 3 9 4 6 9 3 5 1 

correct output
1 9 2 7 4 6 8 3 5 1 

user output
1 9 2 7 4 6 8 3 5 1 

Test 29

Verdict: ACCEPTED

input
10 1
4 6 5 5 1 2 4 2 1 3 

correct output
1 1 1 1 1 1 1 1 1 1 

user output
1 1 1 1 1 1 1 1 1 1 

Test 30

Verdict:

input
100 55
636562060 767928734 906523441 ...

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

user output
33 42 50 34 51 30 50 24 36 38 ...

Test 31

Verdict:

input
100 42
773442532 122816 137572579 324...

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

user output
34 1 7 17 8 13 6 21 10 20 19 3...

Test 32

Verdict:

input
100 44
198730372 27838076 590195590 4...

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

user output
10 1 31 24 26 22 16 17 8 11 39...

Test 33

Verdict:

input
100 56
75940263 760367935 901888417 3...

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

user output
7 45 51 20 9 34 38 54 30 54 1 ...

Test 34

Verdict: ACCEPTED

input
100 97
967034924 587586158 185430194 ...

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

user output
94 59 19 88 79 67 77 64 27 15 ...

Test 35

Verdict:

input
100 23
59249204 934941692 892631472 2...

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

user output
3 23 23 8 12 23 16 4 18 13 23 ...

Test 36

Verdict:

input
100 90
356460601 224848374 881788059 ...

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

user output
24 14 83 4 1 29 8 55 7 49 57 3...

Test 37

Verdict:

input
100 8
244103474 837431431 342493822 ...

correct output
2 7 3 5 7 5 3 6 3 5 1 1 5 3 1 ...

user output
4 5 5 5 5 5 5 5 5 5 2 2 5 5 1 ...

Test 38

Verdict: ACCEPTED

input
100 88
11934038 257096283 933290530 4...

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

user output
1 24 83 40 56 80 23 44 1 62 46...

Test 39

Verdict:

input
100 2
391337048 538883744 535937150 ...

correct output
1 2 2 2 1 1 1 1 1 1 2 1 1 1 1 ...

user output
2 2 2 2 1 2 2 2 1 2 2 2 2 2 2 ...

Test 40

Verdict:

input
200 110
636562060 767928734 906523441 ...

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

user output
68 87 101 69 103 61 102 48 73 ...

Test 41

Verdict:

input
200 84
773442532 122816 137572579 324...

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

user output
70 1 14 30 16 24 11 38 18 37 3...

Test 42

Verdict:

input
200 88
198730372 27838076 590195590 4...

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

user output
15 2 54 42 45 39 27 28 11 16 6...

Test 43

Verdict:

input
200 111
75940263 760367935 901888417 3...

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

user output
9 86 101 36 15 63 72 107 54 10...

Test 44

Verdict: ACCEPTED

input
200 194
967034924 587586158 185430194 ...

correct output
185 114 39 173 151 128 147 124...

user output
185 114 39 173 151 128 147 124...

Test 45

Verdict:

input
200 45
59249204 934941692 892631472 2...

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

user output
5 41 39 9 18 44 23 6 29 19 34 ...

Test 46

Verdict: ACCEPTED

input
200 179
356460601 224848374 881788059 ...

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

user output
54 33 162 8 4 61 15 113 14 99 ...

Test 47

Verdict:

input
200 16
244103474 837431431 342493822 ...

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

user output
13 16 16 16 16 16 15 16 14 16 ...

Test 48

Verdict:

input
200 175
11934038 257096283 933290530 4...

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

user output
3 41 165 67 105 157 38 77 3 11...

Test 49

Verdict:

input
200 3
391337048 538883744 535937150 ...

correct output
2 2 2 2 1 1 1 1 1 1 3 2 1 1 1 ...

user output
3 3 3 3 1 3 3 3 1 3 3 3 3 3 3 ...

Test 50

Verdict:

input
500 275
636562060 767928734 906523441 ...

correct output
176 215 251 178 254 161 252 12...

user output
180 217 255 182 258 165 256 12...

Test 51

Verdict:

input
500 209
773442532 122816 137572579 324...

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

user output
165 1 33 76 40 61 24 94 48 93 ...

Test 52

Verdict:

input
500 218
198730372 27838076 590195590 4...

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

user output
47 9 135 106 118 99 74 77 39 4...

Test 53

Verdict:

input
500 276
75940263 760367935 901888417 3...

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

user output
22 211 254 87 34 152 168 269 1...

Test 54

Verdict:

input
500 484
967034924 587586158 185430194 ...

correct output
469 290 97 438 372 322 367 317...

user output
470 291 98 439 373 323 368 318...

Test 55

Verdict:

input
500 111
59249204 934941692 892631472 2...

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

user output
8 103 98 25 42 109 55 10 68 46...

Test 56

Verdict:

input
500 447
356460601 224848374 881788059 ...

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

user output
155 89 395 31 18 169 47 280 44...

Test 57

Verdict:

input
500 39
244103474 837431431 342493822 ...

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

user output
27 39 39 39 39 39 39 39 33 39 ...

Test 58

Verdict:

input
500 437
11934038 257096283 933290530 4...

correct output
7 101 407 167 251 388 97 191 7...

user output
7 105 407 171 253 389 101 195 ...

Test 59

Verdict:

input
500 6
391337048 538883744 535937150 ...

correct output
3 4 4 4 1 2 3 2 1 2 6 3 2 2 2 ...

user output
2 2 2 2 2 2 2 2 2 2 6 2 2 2 2 ...