4. 寻找两个有序数组的中位数

题目描述:

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。

请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

nums1 = [1, 3]
nums2 = [2]

则中位数是 2.0

示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

则中位数是 (2 + 3)/2 = 2.5

算法1：（这种方法超过题目要求的时间复杂度了）

使用C++ STL 中的 priority_queue 构建两个堆，一个是最大堆，一个是最小堆。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        priority_queue<int, vector<int>, less<int>> q1;
        priority_queue<int, vector<int>, greater<int>> q2;
        if(nums1.size() == 0 && nums2.size()==0)
            return 0.0;
        for(int i=0; i<nums1.size(); i++)
        {
            fun(i, q1, q2, nums1);
        }
        for(int i=0; i<nums2.size(); i++)
        {
            fun(i, q1, q2, nums2);
        }
        if(((q1.size()+q2.size()) & 1) == 1)
        {
            return (double)q2.top();
        }
        else
            return (double)(q1.top() + q2.top())/2;
        
    }
    void fun(int i, priority_queue<int, vector<int>, less<int>> &q1, priority_queue<int, vector<int>, greater<int>> &q2,vector<int>& nums)
    {
        if(((q1.size()+q2.size()) & 1) == 1)
        {
            if(nums[i] > q2.top())
            {
                q1.push(q2.top());
                q2.pop();
                q2.push(nums[i]);
            }
            else
                q1.push(nums[i]);
        }
        else
        {
            if(q2.empty())
                q2.push(nums[i]);
            else
            {
                if(nums[i] < q1.top())
                {
                    q2.push(q1.top());
                    q1.pop();
                    q1.push(nums[i]);
                }
                else
                    q2.push(nums[i]);
            }
        }
    }
};

算法2： 

想到好的优化方法，再补充