XTU1262 优先队列

题解

使用优先队列模拟一下就可以了，但是需要注意进行优化，题目有b=0的情况，也就是说无论如何捞鱼都不会变，这种情况需要直接优化掉，不能进行模拟，否则会超时。

代码

#include <bits/stdc++.h>
#define LL long long
#define UP(i,l,h) for(int i=l;i<h;i++)
#define DOWN(i,h,l) for(int i=h-1;i>=l;i--)
#define W(t) while(t)
#define MEM(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define COUT(x) cout<<x<<endl
using namespace std;

struct Node{
    int a,b;
    Node(int a,int b):a(a),b(b){}
    Node(){}
    bool operator < (const Node nb) const{
        return a<nb.a;
    }
};
priority_queue<Node> pq;
int main(){
    int t;
    scanf("%d",&t);
    W(t--){
        W(!pq.empty()) pq.pop();
        int n,m;
        scanf("%d%d",&n,&m);
        UP(i,0,n){
            Node nd;
            scanf("%d%d",&nd.a,&nd.b);
            pq.push(nd);
        }
        int ans=0;
        W(!pq.empty()&&(m--)){
            Node nd=pq.top();
            ans+=nd.a;
            pq.pop();
            if(nd.b==0){
                ans+=m*nd.a;
                break;
            }
            if(nd.a-nd.b>0) pq.push(Node(nd.a-nd.b,nd.b));
        }
        printf("%d\n",ans);
    }
}