7-53 两个有序序列的中位数

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

#include<bits/stdc++.h>
using namespace std;

int findMedianSortedArrays(vector<int>nums1, vector<int>nums2) 
{
        int num = 0; // 偶数中位数数字，奇数中位数右侧数字
        int len = nums1.size() + nums2.size();
        bool b = len % 2 == 0; // 是否为偶数
        len /= 2; // 偶数中位数位置，奇数中位数右侧位置
        if (nums1.size() + nums2.size() == 0) return 0; // 空数组返回0
        if (nums1.size() == 0) return b ? (nums2[len - 1] + nums2[len]) / 2.0 : nums2[len]; // 数组1为空返回数组2中位数
        if (nums2.size() == 0) return b ? (nums1[len - 1] + nums1[len]) / 2.0 : nums1[len]; // 数组2为空返回数组1中位数
        int i = 0;
        int j = 0;
        int oldNum = num; // 奇数中位数左侧数字
        while (i + j != len + 1) { // 判断是否循环至中位数（数组合并+数组排序+计数）
            oldNum = num;
            if (i >= nums1.size() || (j < nums2.size() && nums1[i] > nums2[j])) num = nums2[j++];
            else num = nums1[i++];
        }
        return b ? (num + oldNum) / 2.0 : num; // 返回中位数
}

int main()
{
    int num;
    cin>>num;
    vector<int>v1(num),v2(num);
    for(int i=0;i<num;i++)
    {
        cin>>v1[i];
    }
        for(int i=0;i<num;i++)
    {
        cin>>v2[i];
    }
    cout<<findMedianSortedArrays(v1,v2);
}