7-53 两个有序序列的中位数 (25分)

已知有两个等长的非降序序列S1, S2, 设计函数求S1与S2并集的中位数。有序序列A0,A1,⋯,A**N−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

有序列表的中位数,可以采用链表进行弹出

#include <iostream>
#include <deque>
using namespace std;
int main() {
    int N;
    scanf("%d", &N);
    deque<int> d1(N), d2(N);
    for(int i = 0; i < N; i++)
        scanf("%d", &d1[i]);
    for(int i = 0; i < N; i++)
        scanf("%d", &d2[i]);
    int ans;
    while(N--) {
        if(d1.size() != 0 && d2.size() != 0) {
            if(d1[0] < d2[0]) {
                ans = d1.front();
                d1.pop_front();
            }
            else {
                ans = d2.front();
                d2.pop_front();
            }
        } else if(d1.size() != 0) {
            ans = d1.front();
            d1.pop_front();
        }
        else {
            ans = d2.front();
            d2.pop_front();
        }
    }
    printf("%d\n", ans);
    return 0;
}
posted @ 2020-06-14 15:18  SteveYu  阅读(385)  评论(0编辑  收藏  举报