7-53 两个有序序列的中位数 (25分)
已知有两个等长的非降序序列S1, S2, 设计函数求S1与S2并集的中位数。有序序列A0,A1,……AN-1 的中位数指A(N−1)/2 的值,即第⌊(N+1)/2⌋个数(A0为第1个数)。
输入格式:
输入分三行。第一行给出序列的公共长度N(0<N≤100000),随后每行输入一个序列的信息,即N个非降序排列的整数。数字用空格间隔。
输入样例1:
输出格式:
在一行中输出两个输入序列的并集序列的中位数。
输入样例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
题目分析:
①.此题所谓两序列并集不能去掉重复数字!!!否则最后一个测试点WA!!!
②.此题为两个等长数组的排列,合并后的数组有偶数个数字,由于在C中奇数/2会取整,导致最后所求的中位数的下标(2N-1)/2=N-1。那么此时,取最极端情况,就算一个有序数列的最小值大于另一个有序数列的最大值,中位数也会落在较小的数列的最后一个值,也就是说,对两个序列合并的时候,不存在一个数列的值已经完全按升序合并入新数列,但中位数还没有找到的情况。
思路分析
用a[n],b[n]两个数组将两个有序数列储存。
对每个位置的数字依次进行排序,即首先a[0]与b[0]比较大小,将小的数字储存在新数列c[n]中,再将小的数字所在数列的下标加1进行比较。比如:a[0]>b[0],那么c[0]=b[0],再比较b[1]与a[0]……
当然比较的前提是a,b两数组下标都小于n,如果有一个数组已经按升序合并入新数列,那么中位数已经找到。所以我们只需要在a,b两数组下标都小于n的情况下比较。
也就是说,只要合并出来的新数列C[n]的数组下标已经大于N-1,则可以停止比较。
#include<stdio.h>
int main()
{
int sum;
int i;
int num;
scanf("%d",&sum);//等长序列的长度
int a[sum],b[sum],c[2*sum];
for(i=0;i<sum;i++)
{
scanf("%d",&a[i]);
}
for(i=0;i<sum;i++)
{
scanf("%d",&b[i]);
}
int len = 0;
int j = 0,k=0;
while(j <sum && k<sum)
{
if(a[j]>b[k])
{
c[len]=b[k];
len++;
k++;
}
else if(a[j]<b[k])
{
c[len]=a[j];
len++;
j++;
}
else
{
c[len]=a[j];
len++;
j++;
c[len]=b[k];
len++;
k++;
}
if((len-1)>=(2*sum-1)/2)//只要合并出来的新数列C[n]的数组下标已经大于N-1,则可以停止比较**。
break;
}
printf("%d",c[(2*sum-1)/2]);
return 0;
}