进阶实验1-3.1 两个有序序列的中位数 (25 分)

进阶实验1-3.1 两个有序序列的中位数 (25 分)

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

方法一：两个序列不断比较增加，当取到第N个数的时候即为中位数，此时的时间复杂度是O(N)，空间复杂度O(1)。

#include <stdio.h>

int main()
{
    int i,n,a[100000]={},b[100000]={};
    scanf("%d",&n);
    
    for(i=0;i<n;i++)
        scanf("%d",&a[i]);
    for(i=0;i<n;i++)
        scanf("%d",&b[i]);
    
    int j=0,z=0;
    for(i=0;i<n-1;i++)
    {
        if(a[j]>b[z]) z++;
        else j++;
    }
    if(a[j]>b[z]) printf("%d",b[z]);
    else printf("%d",a[j]);
}

方法二：利用二分法，将两个序列不断取中位数，并获取新序列，最后得到原数列的中位数。此时，时间复杂度为O(log N)，空间复杂度为O(1)。

#include <stdio.h>

int main()
{
    int i,n,a[100000]={},b[100000]={};
    scanf("%d",&n);
    
    for(i=0;i<n;i++)
        scanf("%d",&a[i]);
    for(i=0;i<n;i++)
        scanf("%d",&b[i]);
    
    int mid1,mid2;
    int al=0,bl=0;
    int ar=n-1,br=n-1;
    while(ar-al+1!=1)
    {
        mid1=(ar+al)/2;
        mid2=(br+bl)/2;
        
        if(a[mid1]>b[mid2]) 
        {
            ar=mid1;
            if((br-bl+1)%2==0) bl=mid2+1;
            else bl=mid2;
        }
        else if(a[mid1]<b[mid2])
        {
            br=mid2;
            if((ar-al+1)%2==0) al=mid1+1;
            else al=mid1;
        }
        else break;
        
    }
    mid1=(ar+al)/2;
    mid2=(br+bl)/2;
    
    if(a[mid1]>b[mid2]) printf("%d",b[mid2]);
    else printf("%d",a[mid1]);
}