hdu4544湫湫系列故事——消灭兔子 priority queue

原创于 2014-09-24 00:19:18 发布 · 527 阅读

0 ·

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

标签

#stl

收录于

stl 专栏收录该内容

2 篇文章

订阅专栏

本文通过一个C++程序示例介绍了如何使用小顶堆解决特定问题：为一系列具有不同射程和价值的箭矢分配目标，确保每个箭矢都能命中目标且总价值最大。代码中展示了小顶堆的实现方法、箭矢排序策略及主要逻辑。

#include <iostream>
#include <cstdio>
#include <queue>
#include <algorithm>
using namespace std;
const int N=100009;
struct node{
    int d,p;
    friend bool operator <(const node &a,const node &b){
        return a.p>b.p;//小顶堆,
    }
}r[N];
bool cmp(node a,node b){
    return a.d>b.d;
}
priority_queue<node> q;
int b[N],n,m;
int main(){
    while(scanf("%d%d",&n,&m)==2){
        while(!q.empty())q.pop();
        for(int i=0;i<n;i++) scanf("%d",b+i);
        for(int i=0;i<m;i++)scanf("%d",&r[i].d);
        for(int i=0;i<m;i++){
            scanf("%d",&r[i].p);
        }
    long long ans=0;
    sort(b,b+n);
    sort(r,r+m,cmp);
    //for(int i=0;i<n;i++)cout<<b[i]<<endl;
    int cnt=0;
    bool flag=true;
    for(int i=n-1;i>=0;i--){
        while(cnt<m&& r[cnt].d>=b[i]){//符合条件的才要
            q.push(r[cnt]);cnt++;
        }
        if(q.empty()){
            flag=false;break;
        }
        ans+=q.top().p;
        q.pop();//每种箭只能使用一次
    }
    if(!flag)cout<<"No"<<endl;
    else cout<<ans<<endl;
    }
    return 0;
}