003-寻找两个有序数组的中位数

题目

给定两个大小为 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

解题思路

自己想的笨办法，效率不是很高。

//
// Created by HINTS on 2019/3/12.
//findMedianSortedArrays


#include <iostream>
#include <set>
#include <vector>
using namespace std;

double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
    //利用multiset将两个数组合并
    multiset<int> num;
    double res = 0;
    for (int i = 0; i < nums1.size(); ++i) {
        num.insert(nums1[i]);
    }
    for (int j = 0; j < nums2.size(); ++j) {
        num.insert(nums2[j]);
    }
    int len = num.size();
    set<int >::iterator set_it = num.begin();
    //while (set_it != num.end()){
    //    cout << *set_it << endl;
    //    set_it++;
    //}
    //通过遍历查找合并数组的中位数
    if(len % 2 ==1){
        for(int i = 0; i < len/2+1; i++){
            res = *set_it;
            set_it++;
        }
        return res;
    } else{
        for(int i = 0; i < len/2; i++){
            res = *set_it;
            set_it++;
        }

        return  (res + *set_it)/2.0;
    }

}

int main(){
    vector<int> nums1 = {1, 2}, nums2 = {3, 4};
    cout << findMedianSortedArrays(nums1, nums2)<<endl;
    return 0;
}