算法实践：寻找两个有序数组的中位数（归并）

leetcode4 寻找两个有序数组的中位数

描述

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

难度

中，归并排序

解题思路

利用归并排序的思想，数组的总个数为偶数时，寻找第(l1+l2)/2和(l1+l2)/2+1个数 。为奇数时，寻找第 (l1+l2)/2+1个数。

代码

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int l1 = nums1.length;
        int l2 = nums2.length;
        int k=0,a=0,b=0,count=0;
        int i=0,j=0;
        if((l1+l2)%2==0){
            while(i<l1 && j<l2){
                if(nums1[i]<=nums2[j]){
                    count++;
                    i++;
                    if(count==(l1+l2)/2) a = nums1[i-1];
                    if(count==(l1+l2)/2+1) b = nums1[i-1];
                }else{
                    count++;
                    j++;
                    if(count==(l1+l2)/2) a = nums2[j-1];
                    if(count==(l1+l2)/2+1) b = nums2[j-1];                    
                }
                if(a!=0 && b!=0)  return((a+b)/2.0);               
            }
            while(i<l1){
                count++;
                i++;
                if(count==(l1+l2)/2) a = nums1[i-1];
                if(count==(l1+l2)/2+1) b = nums1[i-1];
            }
            while(j<l2){
                count++;
                j++;
                if(count==(l1+l2)/2) a = nums2[j-1];
                if(count==(l1+l2)/2+1) b = nums2[j-1];
            }
            return((a+b)/2.0); 
        }
        else{
            while(i<l1 && j<l2){
                if(nums1[i]<=nums2[j]){
                    count++;
                    i++;
                    if(count==(l1+l2)/2+1) k = nums1[i-1];
                }else{
                    count++;
                    j++;
                    if(count==(l1+l2)/2+1) k = nums2[j-1];        
                }
                if(k!=0)  return(k);               
            }
            while(i<l1){
                count++;
                i++;
                if(count==(l1+l2)/2+1) k = nums1[i-1];
            }
            while(j<l2){
                count++;
                j++;
                if(count==(l1+l2)/2+1) k = nums2[j-1];
            }
            return k;
        }
    }
}