给定有序序列 a,b,求出所有a[i]+b[j]中所的第k小的数

算法:
a[1]+b[1]一定是最小的,退出;
a[1]+b[2],a[2]+b[1]放入堆,得到一个小根堆
退出根a[x]+b[y],进入一个元素a[x]+b[y+1],如果y==1再进入a[x+1]+b[y]
重复上面步骤,退出k个数就得到最小的k个数(复杂度O(klogk))


因为退出的是堆的根,假设是a[x]+b[y],
如果在剩下的数中还存在一个数比a[x]+b[y+1]小,设是a[x']+b[y']
a[x]+b[y] < a[x']+b[y'] < a[x]+b[y+1]
那么有3中情况
1、x'<x
那么y'>y
a[x']+b[y']<=a[x-1]+b[y+1],且a[x]+b[y] > a[x-1]+b[y],所以在a[x]+b[y]退出之前a[x-1]+b[y]一定已经退出
在a[x-1]+b[y]退出时,a[x-1]+b[y+1]已经加入堆中,所以a[x-1]+b[y+1]一定不在剩下的数中。
2、x'==x 不存在
3、x'>x
a[x+1]+b[y]>=a[x']+b[y']>=a[x+1]+b[1],


    若y==1,a[x+1]+b[1]还未加入到堆中,同时加入a[x+1]+b[1]、a[x]+b[2];
    若y>1,a[x+1]+b[1]要么在堆中,要么已经退出。
    若在堆中,则a[x+1]+b[2]一定不在堆中,
            如果a[x+1]+b[1] > a[x]+b[y+1] 那么a[x+1]+b[2] > a[x]+b[y+1],扔应该加入a[x]+b[y+1]
            如果a[x+1]+b[1] < a[x]+b[y+1] 加入a[x]+b[y+1]之后,根一定会被调整为a[x+1]+b[1],能确保a[x+1]+b[2]在a[x]+b[y+1]退出之前被加入到堆
    y'=2~y都可以以此类推

所以按照上述规则键堆,每次退出的一定是当前最小的元素

#include<iostream>
#include<queue>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
class fr
{
public:
	int x,y,v;
    bool operator <(const fr &t) const
    {
        return v<t.v;
    }
    bool operator >(const fr &t) const
    {
        return v>t.v;
    }
};

priority_queue<fr,vector<fr> ,greater<fr> > q;

int a[1000],b[1000];
int t[1000000];
	int n,m,k;
int key;

void check()
{
	int size=0;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			t[size++]=a[i]+b[j];
    sort(t,t+size);
    cout<<t[k-1]<<endl;
}

int main()
{

	while(scanf("%d%d%d",&n,&m,&k)!=-1)
	{
		while(q.size())
			q.pop();
		int i;
		for(i=1;i<=n;i++)
			scanf("%d",a+i);
		for(i=1;i<=m;i++)
			scanf("%d",b+i);
		fr tep,tt;
		if(k == 1)
		{
			cout<<a[1]+b[1]<<endl;
			continue;
		}
		int gap=k-1;
		tep.x=2;
		tep.y=1;
		tep.v=a[2]+b[1];
		q.push(tep);
		swap(tep.x,tep.y);
		tep.v=a[1]+b[2];
		q.push(tep);
		while(gap--)
		{
			tep=q.top();
            key=tep.v;
			q.pop();
			tt=tep;
			tt.y++;
			if(tt.y<=m)
			{
				tt.v=a[tt.x]+b[tt.y];
				q.push(tt);
			}
			if(tep.y==1)
			{
				tt.y=1;
			    tt.x=tep.x+1;
				if(tt.x<=n)
				{
                    	tt.v=a[tt.x]+b[tt.y];
				     q.push(tt);
				}
			}

		}
		cout<<key<<endl;
		check();
	}
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值