Codeforces #488div.2 - 994B - Knights of a Polygonal Table(堆+贪心)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Myriad_Dreamin/article/details/80716470

很明显要维护一个大小为k的小顶堆，每次判断小顶堆是否值得被剔除。
需要注意两点：
①力量相同的骑士共用一个堆状态；
②堆为0时可能会导致取空堆的情况；
③10*10^9>INTMAX

#include <cstdio>
#include <queue>
#include <algorithm>
using namespace std;
struct ap{int k,b,id;}a[100005];
bool cmp(ap x,ap y){return x.k<y.k;}
long long ansx[100005];
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i].k);
        a[i].id=i;
    }
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i].b);
    }
    sort(a+1,a+1+n,cmp);
    priority_queue<int,vector<int>,greater<int> >Q;
    long long ans=0,ansp=0;
    for(int i=1;i<=n;i++){
        if(a[i].k!=a[i-1].k)ansp=ans;
        ansx[a[i].id]=ansp+a[i].b;
        if(m&&Q.size()==m&&Q.top()<a[i].b){
            ans-=Q.top();Q.pop();
            ans+=a[i].b;Q.push(a[i].b);
        }
        if(Q.size()<m){
            Q.push(a[i].b);
            ans+=a[i].b;
        }
    }
    for(int i=1;i<=n;i++){
        printf("%I64d ",ansx[i]);
    }
}