leetcode Median of Two Sorted Arrays

Median of Two Sorted Arrays

题目详情：

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

Example 1:

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

The median is 2.0

Example 2:

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

The median is (2 + 3)/2 = 2.5

解题方法：

看到两个有序数组，要找他们两个的中位数，首先就要想到把这两个数组进行排序，要求算法的复杂度不超过log（m+n），自然就想到了归并排序，归并排序的复杂度是nlogn，而这次只需将用到一次合并，所以能满足算法时间复杂度的要求。设l1为第一个数组，l2为第二个数组，l3为新开辟的大小为l1和l2的大小之和的数组。i = 0，j = 0。i,j分别代表l1的下标，l2的下标，当i,j都分别小于l1的大小，l2的大小时，入过l1[i]<=l2[j],则i=i+1，将l1[i]放入新开辟的数组l3中；否则j = j+1,将l2[j]放入新开辟的数组l3中。如果循环过后，i仍然小于l1的大小，则将l1中未放入l3的数都放入到l3中。同理若j仍然小于l2的大小,则将l2中未放入l3的数都放入l3中，l3就是l1和l2排序后的数组，再找其中位数就可以了。

详细代码：

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int i = 0; // nums1的下标
        int j = 0; // nums2的下标
        vector<int> temp; // 临时数组，用于存放nums1和nums2排序后合并的数组
        double result = 0;
        while (i < nums1.size() && j < nums2.size()) {
            if (nums1[i] <= nums2[j]) {
                temp.push_back(nums1[i]);
                i++;
            } else {
                temp.push_back(nums2[j]);
                j++;
            }
        }
        while (i < nums1.size()) {
            temp.push_back(nums1[i]);
            i++;
        }
        while (j < nums2.size()) {
            temp.push_back(nums2[j]);
            j++;
        }
        if (temp.size()%2 == 0) {
            result = (double)(temp[temp.size()/2] + temp[temp.size()/2 - 1])/2;
        } else {
            result = temp[temp.size()/2];
        }
        return result;
    }
};