一个长度为L的升序序列S,处在[L/2]个位置的数称为S的中位数。例如,若序列S1=(11,13,15,17,19),则S1的中位数是15,两个序列的中位数是含它们所有元素的升序序列的中位数。例如若S2=(2,4,6,8,20),则S1和S2的的中位数是11。现在有两个等长的升序序列A和B,找出两个序列的中位数。
基本思路:将AB按升序合并后第n个元素位置即为中位数。实际上,只需排序前n个元素即可,因为AB均是升序排列的所以排序很简单,将a中第一个元素跟b第一个元素比较,较小的那个填入序列c中,若a更小,接下来又将a中第二个元素跟b第一个元素比较,较小的那个填入序列c中,以此类推直至c中有n个元素。c的第n个元素即为中位数。
#include <stdio.h>
int find_median(int a[],int b[],int n) //寻找a、b合并后升序序列的中位数
{
int c[n];
int pos_A=0;
int pos_B=0;
for(int i=0;i<n;i++)
{
if(a[pos_A]<=b[pos_B])
{
c[i]=a[pos_A];
pos_A++;
}
else{
c[i]=b[pos_B];
pos_B++;
}
}
return c[n-1]; //返回中位数的值
}
int main(){
int n=5;
int a[5]={11,13,15,17,19};
int b[5]={2,4,6,8,20};
printf("数组a为:\n");
for(int i=0;i<n;i++)
{
printf("%d ",a[i]);
}
printf("\n");
printf("数组b为:\n");
for(int i=0;i<n;i++)
{
printf("%d ",b[i]);
}
printf("\n两个序列的中位数为:%d", find_median(a,b,n));
return 0;
}
运行结果:
优化:事实上,只需得到中位数,甚至都不需要定义数组c,直接定义一个变量d,用d代替上述代码数组c的所有位置,最后输出的d就是中位数。这样优化可以使得空间复杂度由原本的O(n)优化为O(1)。