제출 #1060513

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1060513		Jarif_Rahman	Gift (IZhO18_nicegift)	C++17	7 / 100	789 ms	220768 KiB

nicegift

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

    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;
        for(int x: cur) A[x]-=X-lim;
        ans.push_back({X-lim, cur});
        X = lim;
        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";
    }
}

컴파일 시 표준 에러 (stderr) 메시지

nicegift.cpp: In function 'int main()':
nicegift.cpp:24: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]
   24 |     while(pq.size() >= k){
      |           ~~~~~~~~~~^~~~

• Subtask 1: 7 / 7
• Subtask 2: 0 / 11
• Subtask 3: 0 / 12
• Subtask 4: 0 / 19
• Subtask 5: 0 / 51