[code] PTA 胡凡算法笔记 DAY034

题意 A1029 Median

• 题意
  输入两个序列，输出两个序列合并之后的中位数（偶数取前面）。

• 思路
  这题其实和链表合并的思路很像，这里我是将输入的数据先进行排序，然后分别用i,j标识扫描到的位置，中位数的位置我是采用从1开始的所以是四舍五入的方式，然后循环，因为要考虑有一方过长的情况，所以循环完之后还需要处理。小结处有更简洁的实现方式。
  注意:存在一个序列过长中位数在其中的情况。

• Code in C++

#include <cstdio>
#include <algorithm>

#define maxn 200001

long first[maxn], second[maxn];

int main()
{
    int n, m;
    scanf("%d", &n);
    for (int i = 0; i < n; ++i)
    {
        scanf("%ld",&first[i]);
    }
    std::sort(first, first + n);
    scanf("%d", &m);
    for (int i = 0; i < m; ++i)
    {
        scanf("%ld",&second[i]);
    }
    std::sort(second, second + m);
    // 四舍五入
    long key = (n+m)/2;
    if ((n+m) % 2)
    {
        ++key;
    }
    int i,j;
    for (i = 1, j = 1; i <= n && j <= m;)
    {
        if (first[i-1] <= second[j-1])
        {
            if (i+j-1 == key)
            {
                printf("%ld",first[i-1]);
                return 0;
            }
            ++i;
        }
        else
        {
            if (i+j-1 == key)
            {
                printf("%ld",second[j-1]);
                return 0;
            }
            ++j;
        }
    }

    if (i<=n)
    {
        printf("%ld",first[key+1-i-j]);
    }
    else if (j<=m)
    {
        printf("%ld",second[key+1-i-j]);
    }
    return 0;
}

小结

这题里解析里面有用一种为数组最后一个数字设置最大值的方式来简化处理方式，即只用循环不用最后循环完之后再判断是否存在一方没循环完的情况。

S1[n] = S2[m] = INF;
int i = 0, j = 0, count = 0;
while (count < medianPos)
{
	if (S1[i] < S2[j]) ++i;
	else ++j;
	++count;
}