Task: | Good grades |
Sender: | aalto2024j_001 |
Submission time: | 2024-11-06 17:11:59 +0200 |
Language: | C++ (C++20) |
Status: | READY |
Result: | ACCEPTED |
test | verdict | time | |
---|---|---|---|
#1 | ACCEPTED | 0.00 s | details |
#2 | ACCEPTED | 0.00 s | details |
#3 | ACCEPTED | 0.00 s | details |
#4 | ACCEPTED | 0.00 s | details |
#5 | ACCEPTED | 0.00 s | details |
#6 | ACCEPTED | 0.00 s | details |
#7 | ACCEPTED | 0.00 s | details |
#8 | ACCEPTED | 0.00 s | details |
#9 | ACCEPTED | 0.00 s | details |
#10 | ACCEPTED | 0.00 s | details |
#11 | ACCEPTED | 0.00 s | details |
#12 | ACCEPTED | 0.00 s | details |
#13 | ACCEPTED | 0.00 s | details |
#14 | ACCEPTED | 0.00 s | details |
#15 | ACCEPTED | 0.00 s | details |
#16 | ACCEPTED | 0.00 s | details |
#17 | ACCEPTED | 0.00 s | details |
#18 | ACCEPTED | 0.01 s | details |
#19 | ACCEPTED | 0.00 s | details |
#20 | ACCEPTED | 0.00 s | details |
#21 | ACCEPTED | 0.00 s | details |
#22 | ACCEPTED | 0.01 s | details |
#23 | ACCEPTED | 0.01 s | details |
#24 | ACCEPTED | 0.00 s | details |
#25 | ACCEPTED | 0.00 s | details |
#26 | ACCEPTED | 0.00 s | details |
#27 | ACCEPTED | 0.00 s | details |
#28 | ACCEPTED | 0.00 s | details |
#29 | ACCEPTED | 0.00 s | details |
#30 | ACCEPTED | 0.00 s | details |
#31 | ACCEPTED | 0.00 s | details |
#32 | ACCEPTED | 0.00 s | details |
#33 | ACCEPTED | 0.00 s | details |
#34 | ACCEPTED | 0.00 s | details |
#35 | ACCEPTED | 0.00 s | details |
#36 | ACCEPTED | 0.00 s | details |
#37 | ACCEPTED | 0.00 s | details |
#38 | ACCEPTED | 0.00 s | details |
#39 | ACCEPTED | 0.00 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.00 s | details |
#50 | ACCEPTED | 0.03 s | details |
#51 | ACCEPTED | 0.03 s | details |
#52 | ACCEPTED | 0.03 s | details |
#53 | ACCEPTED | 0.03 s | details |
#54 | ACCEPTED | 0.03 s | details |
#55 | ACCEPTED | 0.02 s | details |
#56 | ACCEPTED | 0.03 s | details |
#57 | ACCEPTED | 0.01 s | details |
#58 | ACCEPTED | 0.03 s | details |
#59 | ACCEPTED | 0.01 s | details |
Compiler report
input/code.cpp: In function 'int solve()': input/code.cpp:162:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare] 162 | for (int i = 0; i < groups.size(); i++) { | ~~^~~~~~~~~~~~~~~
Code
#ifdef ONPC #define _GLIBCXX_DEBUG #endif #include <bits/stdc++.h> #define char unsigned char #define rep(i, a, b) for(int i=a; i< (b); ++i) #define all(x) begin(x), end(x) #define sz(x) (int)(x).size() #define eb emplace_back #define mp make_pair #define mt make_tuple #define fi first #define se second #define pb push_back #define LSOne(S) ((S) & -(S)) using namespace std; // mt19937 rnd(239); mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count()); template <typename T> int sgn(T x) { return (T(0) < x) - (x < T(0)); } typedef long double T; typedef complex<T> pt; #define X real() #define Y imag() template<class T> istream& operator>> (istream& is, complex<T>& p) { T value; is >> value; p.real(value); is >> value; p.imag(value); return is; } typedef long long ll; typedef long double ld; using pi = pair<ll, ll>; using vi = vector<ll>; template <class T> using pq = priority_queue<T>; template <class T> using pqg = priority_queue<T, vector<T>, greater<T>>; int popcnt(int x) { return __builtin_popcount(x); } int popcnt(ll x) { return __builtin_popcountll(x); } #define MIN(v) *min_element(all(v)) #define MAX(v) *max_element(all(v)) #define LB(c, x) distance((c).begin(), lower_bound(all(c), (x))) #define UB(c, x) distance((c).begin(), upper_bound(all(c), (x))) void __print(int x) {cerr << x;} void __print(long x) {cerr << x;} void __print(long long x) {cerr << x;} void __print(unsigned x) {cerr << x;} void __print(unsigned long x) {cerr << x;} void __print(unsigned long long x) {cerr << x;} void __print(float x) {cerr << x;} void __print(double x) {cerr << x;} void __print(long double x) {cerr << x;} void __print(char x) {cerr << '\'' << x << '\'';} void __print(const char *x) {cerr << '\"' << x << '\"';} void __print(const string &x) {cerr << '\"' << x << '\"';} void __print(bool x) {cerr << (x ? "true" : "false");} template<typename T, typename V> void __print(const pair<T, V> &x) {cerr << '{'; __print(x.first); cerr << ", "; __print(x.second); cerr << '}';} template<typename T> void __print(const T &x) {int f = 0; cerr << '{'; for (auto &i: x) cerr << (f++ ? ", " : ""), __print(i); cerr << "}";} void _print() {cerr << "]\n";} template <typename T, typename... V> void _print(T t, V... v) {__print(t); if (sizeof...(v)) cerr << ", "; _print(v...);} #ifdef DEBUG #define dbg(x...) cerr << "\e[91m"<<__func__<<":"<<__LINE__<<" [" << #x << "] = ["; _print(x); cerr << "\e[39m" << endl; #else #define dbg(x...) #endif template<typename S, typename T = S> void chmin(S &s, T t) {s = s < t ? s : t;} template<typename S, typename T = S> void chmax(S &s, T t) {s = s > t ? s : t;} const int INF = 1e9; // 10^9 = 1B is < 2^31-1 const ll LLINF = 4e18; // 4*10^18 is < 2^63-1 const double EPS = 1e-9; const ll MOD = 1e9+7; const int N=550; ll dp[N][N]; int solve() { int n, k; std::cin >> n >> k; vector<pair<ll, int>> v(n+1); for (int i = 1; i <= n; i++) { std::cin >> v[i].fi; v[i].se=i; } for (int i = 0; i <= n; i++) { for (int j = 0; j <= k; j++) { dp[i][j]=-INF; } } sort(v.begin()+1, v.begin()+n+1); if(false) for (int i = 1; i <= n; i++) { std::cout << v[i].fi << " " << v[i].se << std::endl; } for (int i = 1; i <= n; i++) { dp[i][1]=i*v[1].fi; } for (int currK = 2; currK <= k; currK++) { //std::cout << "w: " << currK << std::endl; for (int i = currK; i <= n; i++) { //std::cout << "for i: " << i << " q: " << currK << " " << i-1 << std::endl; for (int j = i; j >= currK-1; j--) { //std::cout << currK << " " << i << " " << j << std::endl; chmax(dp[i][currK], dp[j-1][currK-1]+(i-j+1)*v[j].fi); } } } if(false) for (int i = 1; i <= k; i++) { std::cout << i << std::endl; for (int j = 1; j <= n; j++) { std::cout << dp[j][i] << " "; } std::cout << std::endl; } //std::cout << "BACK: " << std::endl; vector<vector<int>> groups; int currRight=n; int groupsRem=k; for (int j = n; j>=1; j--) { //std::cout << v[j].fi*(currRight-j+1) << " " << dp[j-1][groupsRem-1] << " " << dp[j][groupsRem] << std::endl; if(v[j].fi*(currRight-j+1)+dp[j-1][groupsRem-1]==dp[currRight][groupsRem]) { //std::cout << "Group " << groupsRem << " found " << j << "-" << currRight << std::endl; vector<int> currG; for (int w = j; w <= currRight; w++) { currG.pb(v[w].se); } groups.pb(currG); groupsRem--; currRight=j-1; } } int currG=k; vector<int> ans(n+1, 1); for (int i = 0; i < groups.size(); i++) { for(auto &w:groups[i]) { ans[w]=currG; //std::cout << w << " "; } currG--; //std::cout << std::endl; } //std::cout << std::endl; for (int i = 1; i <= n; i++) { std::cout << ans[i] << " "; } std::cout << std::endl; return 0; } int32_t main() { ios::sync_with_stdio(0); cin.tie(0); int TET = 1; //cin >> TET; for (int i = 1; i <= TET; i++) { #ifdef ONPC cout << "TEST CASE#" << i << endl; #endif if (solve()) { break; } #ifdef ONPC cout << "__________________________" << endl; #endif } #ifdef ONPC cerr << endl << "finished in " << clock() * 1.0 / CLOCKS_PER_SEC << " sec" << endl; #endif }
Test details
Test 1
Verdict: ACCEPTED
input |
---|
1 1 8 |
correct output |
---|
1 |
user output |
---|
1 |
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: ACCEPTED
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 1 2 3 3 2 1 2 |
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: ACCEPTED
input |
---|
100 55 636562060 767928734 906523441 ... |
correct output |
---|
32 42 50 33 51 29 50 23 35 37 ... |
user output |
---|
32 42 50 33 51 29 50 23 35 37 ... Truncated |
Test 31
Verdict: ACCEPTED
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 8 16 9 13 6 20 10 19 18 3... Truncated |
Test 32
Verdict: ACCEPTED
input |
---|
100 44 198730372 27838076 590195590 4... |
correct output |
---|
9 1 28 21 23 20 15 16 7 10 36 ... |
user output |
---|
9 1 28 21 23 20 15 16 7 10 36 ... Truncated |
Test 33
Verdict: ACCEPTED
input |
---|
100 56 75940263 760367935 901888417 3... |
correct output |
---|
6 44 51 19 8 33 36 54 29 54 1 ... |
user output |
---|
6 44 51 19 8 33 36 54 29 54 1 ... Truncated |
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 ... Truncated |
Test 35
Verdict: ACCEPTED
input |
---|
100 23 59249204 934941692 892631472 2... |
correct output |
---|
2 22 21 6 10 23 12 3 15 11 19 ... |
user output |
---|
2 22 21 6 10 23 12 3 15 11 19 ... Truncated |
Test 36
Verdict: ACCEPTED
input |
---|
100 90 356460601 224848374 881788059 ... |
correct output |
---|
23 14 83 4 1 28 8 54 7 48 56 3... |
user output |
---|
23 14 83 4 1 28 8 54 7 48 56 3... Truncated |
Test 37
Verdict: ACCEPTED
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 |
---|
2 7 3 5 7 5 3 6 3 5 1 1 5 3 1 ... Truncated |
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... Truncated |
Test 39
Verdict: ACCEPTED
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 |
---|
1 2 2 2 1 1 1 1 1 1 2 1 1 1 1 ... Truncated |
Test 40
Verdict: ACCEPTED
input |
---|
200 110 636562060 767928734 906523441 ... |
correct output |
---|
69 86 99 70 101 61 100 47 74 7... |
user output |
---|
69 86 99 70 101 61 100 47 74 7... Truncated |
Test 41
Verdict: ACCEPTED
input |
---|
200 84 773442532 122816 137572579 324... |
correct output |
---|
66 1 14 28 16 23 11 36 18 35 3... |
user output |
---|
66 1 14 28 16 23 11 36 18 35 3... Truncated |
Test 42
Verdict: ACCEPTED
input |
---|
200 88 198730372 27838076 590195590 4... |
correct output |
---|
17 3 53 42 45 40 29 30 13 18 6... |
user output |
---|
17 3 53 42 45 40 29 30 13 18 6... Truncated |
Test 43
Verdict: ACCEPTED
input |
---|
200 111 75940263 760367935 901888417 3... |
correct output |
---|
8 85 100 35 14 62 71 106 51 10... |
user output |
---|
8 85 100 35 14 62 71 106 51 10... Truncated |
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... Truncated |
Test 45
Verdict: ACCEPTED
input |
---|
200 45 59249204 934941692 892631472 2... |
correct output |
---|
3 43 41 11 19 45 25 5 30 21 37... |
user output |
---|
3 43 41 11 19 45 25 5 30 21 37... Truncated |
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 ... Truncated |
Test 47
Verdict: ACCEPTED
input |
---|
200 16 244103474 837431431 342493822 ... |
correct output |
---|
4 14 6 8 13 9 6 10 5 10 2 2 8 ... |
user output |
---|
4 14 6 8 13 9 6 10 5 10 2 2 8 ... Truncated |
Test 48
Verdict: ACCEPTED
input |
---|
200 175 11934038 257096283 933290530 4... |
correct output |
---|
2 40 165 67 105 157 37 77 2 11... |
user output |
---|
2 40 165 67 105 157 37 77 2 11... Truncated |
Test 49
Verdict: ACCEPTED
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 |
---|
2 2 2 2 1 1 1 1 1 1 3 2 1 1 1 ... Truncated |
Test 50
Verdict: ACCEPTED
input |
---|
500 275 636562060 767928734 906523441 ... |
correct output |
---|
176 215 251 178 254 161 252 12... |
user output |
---|
176 215 251 178 254 161 252 12... Truncated |
Test 51
Verdict: ACCEPTED
input |
---|
500 209 773442532 122816 137572579 324... |
correct output |
---|
163 1 31 72 37 57 23 89 45 87 ... |
user output |
---|
163 1 31 72 37 57 23 89 45 87 ... Truncated |
Test 52
Verdict: ACCEPTED
input |
---|
500 218 198730372 27838076 590195590 4... |
correct output |
---|
42 8 130 102 113 97 73 75 34 4... |
user output |
---|
42 8 130 102 113 97 73 75 34 4... Truncated |
Test 53
Verdict: ACCEPTED
input |
---|
500 276 75940263 760367935 901888417 3... |
correct output |
---|
20 205 249 85 32 149 165 267 1... |
user output |
---|
20 205 249 85 32 149 165 267 1... Truncated |
Test 54
Verdict: ACCEPTED
input |
---|
500 484 967034924 587586158 185430194 ... |
correct output |
---|
469 290 97 438 372 322 367 317... |
user output |
---|
469 290 97 438 372 322 367 317... Truncated |
Test 55
Verdict: ACCEPTED
input |
---|
500 111 59249204 934941692 892631472 2... |
correct output |
---|
5 104 100 24 42 109 55 9 70 46... |
user output |
---|
5 104 100 24 42 109 55 9 70 46... Truncated |
Test 56
Verdict: ACCEPTED
input |
---|
500 447 356460601 224848374 881788059 ... |
correct output |
---|
154 88 394 30 18 168 46 279 43... |
user output |
---|
154 88 394 30 18 168 46 279 43... Truncated |
Test 57
Verdict: ACCEPTED
input |
---|
500 39 244103474 837431431 342493822 ... |
correct output |
---|
9 33 13 18 31 19 13 23 11 21 3... |
user output |
---|
9 33 13 18 31 19 13 23 11 21 3... Truncated |
Test 58
Verdict: ACCEPTED
input |
---|
500 437 11934038 257096283 933290530 4... |
correct output |
---|
7 101 407 167 251 388 97 191 7... |
user output |
---|
7 101 407 167 251 388 97 191 7... Truncated |
Test 59
Verdict: ACCEPTED
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 |
---|
3 4 4 4 1 2 3 2 1 2 6 3 2 2 2 ... Truncated |