原题链接
思路
我本来自己想的是对s从小到大的排序,然后再进行枚举,每一个点i只能选取比自己s大的值,因为比自己小的值选起来只会让自己变的更苛刻,有可能s比你小的值里面有比你大的v,但是我们不用管,因为我们已经再这之前枚举过它了,它已经形成了自己的最优。每一个点要选的就是s比自己大的,选s-1个v最大的值就行,但是我不知道用什么数据结构来处理,能logn的时间内找到n个数中的最大的s-1个值,看到有大佬用权值线段树,但是我不会。。。。
看了题解的思路,它是从大到小的枚举s,然后就枚举每一个数,把每一个i的v加进一个小根堆里面,如果i的s的值小于队列中的元素的话,就把队列中的队头取出来(把最小值删除),里面剩下的s个值就是这个i能取到的最大的s-1个v。
#include <bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 1e6 + 10;
const int mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;
using namespace std;
typedef pair<int,int> PII;
typedef pair<ll,ll> PLL;
priority_queue<ll,vector<ll> , greater<ll>>q;
struct Node
{
ll v,s;
}a[N];
bool cmp(Node x,Node y)
{
return x.s>y.s;
}
ll tem,ans;
int main()
{
int n;
cin>>n;
for(int i=1 ; i<=n ; i++) cin>>a[i].v>>a[i].s;
sort(a+1,a+1+n,cmp);
for(int i=1 ; i<=n ; i++)
{
tem+=a[i].v;
q.push(a[i].v);
while(q.size()>a[i].s)
{
tem-=q.top();
q.pop();
}
ans=max(ans,tem);
}
cout<<ans<<endl;
return 0;
}
总结
Topk问题:从n个未排序的数中选出k个最大的数或者最小的数。
解决方案:用优先队列或者multiset。