oj.uz

Submission #1060489

# Submission time Handle Problem Language Result Execution time Memory
1060489 2024-08-15T16:07:53 Z Jarif_Rahman Gift (IZhO18_nicegift) C++17
30 / 100
 841 ms 231524 KB 
#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n, k; cin >> n >> k;
    vector<ll> A(n);
    for(ll &x: A) cin >> x;
    vector<pair<ll, vector<int>>> ans;

    priority_queue<pair<ll, int>> pq;
    for(int i = 0; i < n; i++) pq.push({A[i], i});

    ll sum = 0;
    for(ll x: A) sum+=x;
    if(sum%k != 0 || *max_element(A.begin(), A.end()) > sum/k){
        cout << "-1\n";
        exit(0);
    }

    while(pq.size() >= k){
        vector<int> cur;
        for(int i = 0; i < k; i++) cur.push_back(pq.top().second), pq.pop();
        ll lim = 0;
        if(!pq.empty()) lim = pq.top().first-1;
        ll mn = 1e18+5;
        for(int x: cur) mn = min(mn, A[x]-lim);
        for(int x: cur) A[x]-=mn;
        ans.push_back({mn, cur});
        if(ans.size() > int(3e6)){
            cout << "-1\n";
            exit(0);
        }
        for(int x: cur) if(A[x]) pq.push({A[x], x});
    }

    if(pq.size() != 0){
        cout << "-1\n";
        exit(0);
    }
    cout << ans.size() << "\n";
    for(auto [x, v]: ans){
        cout << x << " ";
        for(int i: v) cout << i+1 << " ";
        cout << "\n";
    }
}

Compilation message

nicegift.cpp: In function 'int main()':
nicegift.cpp:23:21: warning: comparison of integer expressions of different signedness: 'std::priority_queue<std::pair<long long int, int> >::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   23 |     while(pq.size() >= k){
      |           ~~~~~~~~~~^~~~

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB n=4
2 Correct 0 ms 344 KB n=3
3 Correct 0 ms 348 KB n=3
4 Correct 0 ms 344 KB n=4
5 Correct 0 ms 348 KB n=4
6 Correct 0 ms 348 KB n=2

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB n=4
2 Correct 0 ms 344 KB n=3
3 Correct 0 ms 348 KB n=3
4 Correct 0 ms 344 KB n=4
5 Correct 0 ms 348 KB n=4
6 Correct 0 ms 348 KB n=2
7 Correct 0 ms 344 KB n=5
8 Correct 0 ms 348 KB n=8
9 Correct 3 ms 1116 KB n=14
10 Correct 1 ms 860 KB n=11
11 Correct 21 ms 4156 KB n=50000
12 Correct 19 ms 4820 KB n=50000
13 Correct 11 ms 2128 KB n=10
14 Correct 17 ms 3372 KB n=685
15 Correct 18 ms 3836 KB n=623
16 Correct 10 ms 2248 KB n=973

Subtask #3
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB n=4
2 Correct 0 ms 344 KB n=3
3 Correct 0 ms 348 KB n=3
4 Correct 0 ms 344 KB n=4
5 Correct 0 ms 348 KB n=4
6 Correct 0 ms 348 KB n=2
7 Correct 0 ms 344 KB n=5
8 Correct 0 ms 348 KB n=8
9 Correct 3 ms 1116 KB n=14
10 Correct 1 ms 860 KB n=11
11 Correct 21 ms 4156 KB n=50000
12 Correct 19 ms 4820 KB n=50000
13 Correct 11 ms 2128 KB n=10
14 Correct 17 ms 3372 KB n=685
15 Correct 18 ms 3836 KB n=623
16 Correct 10 ms 2248 KB n=973
17 Correct 18 ms 2896 KB n=989
18 Correct 8 ms 984 KB n=563
19 Correct 13 ms 1352 KB n=592
20 Correct 16 ms 1452 KB n=938
21 Correct 10 ms 1236 KB n=747
22 Correct 12 ms 1356 KB n=991

Subtask #4
0 / 19.0

# Verdict Execution time Memory Grader output
1 Incorrect 841 ms 231524 KB Jury has the answer but participant has not
2 Halted 0 ms 0 KB -

Subtask #5
0 / 51.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB n=4
2 Correct 0 ms 344 KB n=3
3 Correct 0 ms 348 KB n=3
4 Correct 0 ms 344 KB n=4
5 Correct 0 ms 348 KB n=4
6 Correct 0 ms 348 KB n=2
7 Correct 0 ms 344 KB n=5
8 Correct 0 ms 348 KB n=8
9 Correct 3 ms 1116 KB n=14
10 Correct 1 ms 860 KB n=11
11 Correct 21 ms 4156 KB n=50000
12 Correct 19 ms 4820 KB n=50000
13 Correct 11 ms 2128 KB n=10
14 Correct 17 ms 3372 KB n=685
15 Correct 18 ms 3836 KB n=623
16 Correct 10 ms 2248 KB n=973
17 Correct 18 ms 2896 KB n=989
18 Correct 8 ms 984 KB n=563
19 Correct 13 ms 1352 KB n=592
20 Correct 16 ms 1452 KB n=938
21 Correct 10 ms 1236 KB n=747
22 Correct 12 ms 1356 KB n=991
23 Incorrect 841 ms 231524 KB Jury has the answer but participant has not
24 Halted 0 ms 0 KB -