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

最新推荐文章于 2024-12-23 16:38:29 发布

原创最新推荐文章于 2024-12-23 16:38:29 发布 · 503 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#codeforces #堆 #贪心

收录于

codeforces 同时被 3 个专栏收录

41 篇文章

订阅专栏

贪心算法

19 篇文章

订阅专栏

堆

1 篇文章

订阅专栏

本文介绍了一种算法实现方法，通过维护一个大小为K的小顶堆来解决特定问题。该算法需要处理骑士的力量值，并确保在操作过程中堆的状态正确更新。文章详细展示了使用C++实现这一算法的过程。

很明显要维护一个大小为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]);
    }
}