B. Lamps

思路

贪心思想,先排序吧序号小的先放进来,如果发现序号一样的话那么就直接按大到小排序来依次选取,多余的部分如何处理?我们可以把每次取得灯泡放入队列里面,如何每次看看队列的对头是不是小于等于当前num,如果小于等于并且队列不空,就直接pop()

#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
using namespace std;
int t, n,x;
struct Node {
    int a, b;
}lst[200005];
bool cmp(Node x, Node y) {
    if (x.a == y.a) return x.b > y.b;//序号一样先大后小
    else return x.a < y.a;//按序号排序
}
int main() {
    cin >> t;
    while (t--) {
        queue<int>q;//定义队列
        cin >> n;
        int num = 0;
        x = 0;
        long long sum = 0;//总和
        for (int i = 1; i <= n; i++) {
            cin >> lst[i].a >> lst[i].b;
        }
        sort(lst + 1, lst + n + 1, cmp);//排序
        for (int i = 1; i <= n; i++)
        {
            if (num < lst[i].a)//当前个数不多,小于目前要放进去的序号的点
            {
                sum += lst[i].b;//加进去
                x++;
                q.push(lst[i].a);
                num=x;//更新一下
            }
            while (!q.empty() && q.front() <= num)//去掉多余
            {
                x--;//取数--
             	q.pop();
             }
             
        }
        
        cout << sum << endl;
    }
    return 0;
}