堆排序可以直接用priority_queue<int >(默认为大顶堆)去直接实现这个大顶堆以及小顶堆,那么我们用一个大顶堆和一个小顶堆,小顶堆中装大于x的数,则小顶堆的堆顶就是大于x的最小值,大顶堆中放比x小的值,则大顶堆的堆顶是小于x的最大值,我们通过两个堆就得到了与x最相近的两个值。
同样的,也可以维护小顶堆的大小,使得小顶堆永远小于某个数K,则这两个堆的作用就是维护所有数中第K大的值,而且是一种动态维护的方法,动态的得到放入的数中第K大的值(或第K小)。
这一题大意就是动态维护第K大的值
所以直接用对顶堆去维护就可以了。
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int a[N];
priority_queue<int> s;
priority_queue<int ,vector<int>,greater<int> > s1;
int main()
{
ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
int n;
int k;
cin>>n>>k;
for(int i=1;i<=n;i++)
{
cin>>a[i];
s1.push(a[i]);
if(!s.empty())
{
s1.push(s.top());
s.pop();
}
int Max=max(1,i*k/100);
while(s1.size()>Max)
{
s.push(s1.top());
s1.pop();
}
cout<<s1.top()<<" ";
}
}
序列合并
这题刚开始做的时候用的贪心取到两个数组前i*j==n个最小值,然后再用一个小顶堆维护这n个数产生的数,但是实际上如果a【1】足够的小,那么b数组是可以取到n-1个的,还是需要考虑所有情况的可能性。我们可以在这个数据中发现一个规律
所以我们需要动态的维护N个数,每次取到N个数中的最小值,然后更新这个值,再放回小顶堆,至于怎么存,其实是可以重载小顶堆的排序规则,使得小顶堆通过我们给出的规则进行维护,也可以用pair代替int存储,再用一个数组去记录这N个序号分别已经指向哪一个a【i】也是可以的。
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1e5+10;
int a[N],b[N],id[N];
priority_queue<pair<int,int>,vector<pair<int,int> >,greater<pair<int,int> > >s;
signed main()
{
ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
int n;
cin>>n;
for(int i=1;i<=n;i++)cin>>a[i];
for(int i=1;i<=n;i++)cin>>b[i];
for(int i=1;i<=n;i++)
{
id[i]=1;
s.push({a[1]+b[i],i});
}
int num=0;
while(!s.empty())
{
//if(num!=0)cout<<" ";
num++;
cout<<s.top().first<<" ";
if(num==n)break;
s.push({b[s.top().second]+a[id[s.top().second]+1],s.top().second});
id[s.top().second]++;
s.pop();
}
}