PAT-A 1029. Median (25)

题目链接在此。

题意

给出两个递增的序列a,b，长度分别为n,m，要求将它们合并成一个新的递增序列后输出中位数。（如果个数为偶数，输出左半部分最后一个数）

思路

首先将两个有序序列合并成一个新的有序序列，之后输出中位数。

AC代码

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;int n,m;long long a[1000010], b[1000010], product[2000010];int main(){    scanf("%d",&n);    for(int i = 0; i < n; i++){        scanf("%lld",&a[i]);    }    scanf("%d",&m);    for(int i = 0; i < m; i++){        scanf("%lld",&b[i]);    }    int i = 0,j = 0;    int index = 0;    while(i < n && j < m){        while( i < n && a[i] < b[j]) product[index++] = a[i++];        while( j < m && a[i] >= b[j]) product[index++] = b[j++];    }    while( i < n) product[index++] = a[i++];    while( j < m) product[index++] = b[j++];    printf("%lld\n",product[(0+index-1)/2]);    return 0;}