7-3 两个有序序列的中位数 (递归的思想)

已知有两个等长的非降序序列S1, S2, 设计函数求S1与S2并集的中位数。有序序列A0,A1,⋯,AN−1的中位数指A(N−1)/2的值,即第⌊(N+1)/2⌋个(A0为第1个数)。

输入格式:
输入分三行。第一行给出序列的公共长度N(0<N≤100000),随后每行输入一个序列的信息,即N个非降序排列的整数。数字用空格间隔。

输出格式:
在一行中输出两个输入序列的并集序列的中位数。

输入样例1:
5
1 3 5 7 9
2 3 4 5 6
输出样例1:
4
输入样例2:
6
-100 -10 1 1 1 1
-50 0 2 3 4 5
输出样例2:
1
先说说思路:

使用递归的思想,首先我们将一个问题变成成若干个子问题,到最后就是函数的出口。

我们以样例1数据为例,令第一个数组为a[n],第二个数组为b[n]

首先找到两个数组各自的中位数下标mid1和mid2,有三种情况:

1、a[mid1]==b[mid2]:  直接输出,它们就是中位数。

2、a[mid1]<b[mid2]:  那么中位数只可能存在在a[mid1]的右边或者b[mid2]的左边(简单地说就是掐头去尾,比如原来找1 2 3 4 5 6 7 8 9这几个数的中位数,我可以变成找4 5 6 7 8的中位数,去掉a数组小的和b数组大的),那我们就只需要在找 a[mid1:n-1]和b[0:mid2]找中位数就可以了,此时我们发现,我们把问题变成了 (找a[mid1:n-1]和b[0:mid2]的中位数),问题规模缩小了一半,不断的缩小就会找到函数的出口。

3、a[mid1]>b[mid2]: 同上,找a[0:mid1]和b[mid2:n-1]的中位数。

代码实现如下:

int mid1=(l1+r1)/2,mid2=(l2+r2+1)/2;//mid2里面为什么要加1后面会说
	if(a[mid1]==b[mid2])
	{
		cout<<a[mid1]<<endl;
		return;
	}
	if(a[mid1]<b[mid2])find(a,mid1,r1,b,l2,mid2);
	if(a[mid1]>b[mid2])find(a,l1,mid1,b,mid2,r2);

求mid2时要加1是考虑到偶数的原因,举个例子:

1 3 5 7

2 4 6 8

3<4,也就是a[mid1]<b[mid2],那么a取到的是3 5 7 ,b取到的是2 4 ,两个数组就不等长了,一直取下去肯定会有bug,加1后就解决了这个问题,对奇数也不会有影响(整除就没了)

最后说一下出口的问题:

出口有两个,一个是到了最后,每个数组只剩两个元素的时候:再缩小下去就是无限循环了,另外一种情况是只剩一个元素,直接比较就行了

int m1,m2;
	if(r1-l1==1){//剩下两个元素的时候,找靠中间的两个(左大右小)中最小那个
		m1=a[l1]<b[l2]?b[l2]:a[l1];
		m2=a[r1]<b[r2]?a[r1]:b[r2];
		if(m1<m2)cout<<m1<<endl;
		else cout<<m2<<endl;
		return;
	}
	if(r1==l1)
	{
		int k=a[r1]<b[r2]?a[r1]:b[r2];
		cout<<k<<endl;
		return;
	}

最后附上总代码

#include <iostream>
using namespace std;
void find(int a[],int l1,int r1,int b[],int l2,int r2)
{
	int m1,m2;
	if(r1-l1==1){
		m1=a[l1]<b[l2]?b[l2]:a[l1];
		m2=a[r1]<b[r2]?a[r1]:b[r2];
		if(m1<m2)cout<<m1<<endl;
		else cout<<m2<<endl;
		return;
	}
	if(r1==l1)
	{
		int k=a[r1]<b[r2]?a[r1]:b[r2];
		cout<<k<<endl;
		return;
	}
	int mid1=(l1+r1)/2,mid2=(l2+r2+1)/2;
	if(a[mid1]==b[mid2])
	{
		cout<<a[mid1]<<endl;
		return;
	}
	if(a[mid1]<b[mid2])find(a,mid1,r1,b,l2,mid2);
	if(a[mid1]>b[mid2])find(a,l1,mid1,b,mid2,r2);
}
int main()
{
	int a[100001],b[100001];
	int n;
	cin>>n;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
	}
	for(int i=0;i<n;i++)
	{
		cin>>b[i];
	}
	find(a,0,n-1,b,0,n-1);
 } 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值