已知有两个等长的非降序序列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);
}