Submission #72369138

---

Source Code Expand

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int M=1e5+5;

int n,k,x;
vector<int> a;
vector<int> b;
priority_queue<pair<int,pair<int,int>>, vector<pair<int,pair<int,int>>>, greater<>> pq;
vector<int> diffs;

signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    
    cin>>n>>k>>x;
    a.resize(n);
    for(int i=0;i<n;i++)cin>>a[i];
    
    sort(a.begin(),a.end(),greater<>());
    int base=k*a[0];
    
    b.resize(n-1);
    for(int i=0;i<n-1;i++)b[i]=a[i]-a[i+1];
    
    diffs.push_back(0);
    if(n>1&&k>0){
        pq.push({b[0],{0,1}});
    }
    
    while(diffs.size()<x&&!pq.empty()){
        auto [val,state]=pq.top();
        pq.pop();
        auto [id,cnt]=state;
        
        diffs.push_back(val);
        if(cnt<k){
            pq.push({val+b[id],{id,cnt+1}});
        }
        
        if(id+1<n-1){
            pq.push({val+b[id+1],{id+1,1}});
        }
    }
    for(int i=0;i<x;i++){
        int idx=min(i,(int)diffs.size()-1);
        cout<<base-diffs[idx]<<'\n';
    }
    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function 'int main()':
./Main.cpp:31:23: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   31 |     while(diffs.size()<x&&!pq.empty()){
      |           ~~~~~~~~~~~~^~

Judge Result