PAT 1029 Median（中位数+归并排序）

题目

https://www.patest.cn/contests/pat-a-practise/1029

题意：求两个排好序的数组合并后的中位数。

解题思路

担心直接sort会超时，所以拿归并排序的merge方法合并数组，即从每次都从两个数组剩余的元素中挑出最小的，若某个数组先挑完，则另一个数组原样抄回。

然而，sort也能过……（摊手）

顺便复习一下归并排序：

归并排序利用了分治的思想，将数列a[l,h]分成两半a[l,mid]和a[mid+1,h]，分别进行归并排序，然后再将这两半合（merge）并起来。

利用归并排序可以求逆序对的数目。即在合并的过程中（设l<=i<=mid，mid+1<=j<=h），当a[i]<=a[j]时，并不产生逆序数；当a[i]>a[j]时，在前半部分中比a[i]大的数都比a[j]大，将a[j]放在a[i]前面时，逆序数要加上mid+1-i。因此，可以在归并排序中的合并过程中计算逆序数。

AC代码

#include <iostream>
using namespace std;

const int maxn = 1000005;
long int a[maxn], b[maxn], res[maxn<<1];
int lena, lenb, cnt;

int main()
{
    ios::sync_with_stdio(false);
    cin >> lena;
    for (int i = 0; i < lena; ++i)
        cin >> a[i];
    cin >> lenb;
    for (int i = 0; i < lenb; ++i)
        cin >> b[i];
    //归并排序的merge步骤
    int p1 = 0, p2 = 0, cnt = 0;
    while (p1 < lena && p2 < lenb)
    {
        if (a[p1] < b[p2]) 
        {
            res[cnt++] = a[p1];
            p1++;
        }
        else
        {
            res[cnt++] = b[p2];
            p2++;
        }
    }
    if (p1 < lena) //p1还有剩余
    {
        for (int i = p1; i < lena; ++i)
            res[cnt++] = a[i];
    }
    if (p2 < lenb) //p2还有剩余
    {
        for (int i = p2; i < lenb; ++i)
            res[cnt++] = b[i];
    }
    cout << res[(cnt-1)>>1] << endl; //输出中位数
    return 0;
}