力扣 题目4- 寻找两个正序数组的中位数

题目

题解

遍历(nums1size + nums2size) / 2 + 1次  nums1与nums2比较谁小取谁  放入vector 

最后根据i与vector 容器的关系返回

我们实际上最多只需要最后遍历的两次 所以可以加入特殊判断 节省空间

代码

#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int nums1size = nums1.size();
        int nums2size = nums2.size();
        int nums1foreach=0;
        int nums2foreach = 0;
        int i = 1;
        int er = (nums1size+ nums2size)%2==0?1:0;
        int zhongwei = (nums1size + nums2size) / 2 + 1;
        int low = 0;
        vector<int> zhong = {};
        for (; i <= zhongwei; i++) {
            if (nums2foreach<nums2size&&(nums1foreach>= nums1size||nums1[nums1foreach]> nums2[nums2foreach])) {
                low = nums2[nums2foreach];
                nums2foreach++;
            }
            else
            {
                low = nums1[nums1foreach];
                nums1foreach++;
            }

            if (i== zhongwei-1|| i == zhongwei) {
                zhong.push_back(low);
            }
        }
        return er? (zhong[0]+ zhong[1])/2.0: zhong.back()*1.0;
    }
};

int main() {
    Solution sol;
    vector<int> nums1 = {1,2};
    vector<int> nums2 = {};
    double result =sol.findMedianSortedArrays(nums1, nums2);
    cout << result << endl;
}