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

思路

定义两个指针i, j，当nums1[i] < nums2[j]时，i++，当nums1[i] > nums2[j]时，j++，同时，通过两个数组的大小，可以确定中位数在第几个，这样一次遍历就可找到位于中位数的元素；

代码

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int len, i1, i2, i = 0, j = 0, sum = 0, count = 0, t = 0;
        len = nums1.length + nums2.length;
        i1 = (len - 1)/2;
        i2 = (len - 1)/2 + (len - 1)%2;
        while (i<nums1.length || j<nums2.length){
            if (i == nums1.length)
                t = nums2[j++];
            else if (j == nums2.length)
                t = nums1[i++];
            else if (nums1[i] < nums2[j])
                t = nums1[i++];
            else if (nums1[i] >= nums2[j])
                t = nums2[j++];
            if (count == i1)
                sum += t;
            if (count == i2)
                sum += t;
            count++;
            if (count > len/2)
                break;
        }
        return sum * 1.0 / 2;
    }
}