PTA 兩個有序序列的中位數(25分)

兩個有序序列的中位數 

 

已知有兩個等長的非降序序列S1, S2, 設計函數求S1與S2並集的中位數。有序序列,的中位數指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

思路：建立一個數組，長度為兩個序列的總和，將序列中所有數字排序並且存入數組中，然后進行判斷。

#include<stdio.h>
int main()
{
    int i,k=0,l=0,n;
    int mid;
    scanf("%d",&n);
    int a[n],b[n],c[2*n];
    for(i=0;i<n;i++){
        scanf("%d",&a[i]);
    }
    for(i=0;i<n;i++){
        scanf("%d",&b[i]);
    }
    i=0;
    while(k<n&&l<n){
        if(a[k]>b[l]){
            c[i++]=b[l++];
        }else{
            c[i++]=a[k++];
        }
    }
    while(k<n){
        c[i++]=a[k++];
    }
    while(l<n){
        c[i++]=a[l++];
    }
    mid=c[(i-1)/2];
    printf("%d",mid);
    return 0;
}