堆排序中的算法

堆排序可以直接用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();
	}
}

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值