LeetCode 4. 尋找兩個正序數組的中位數

LeetCode 4. 尋找兩個正序數組的中位數

題目

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

思路，雙指針 O(n)複雜度

代碼

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int len =nums1.length;
        len+=nums2.length;
        int [] list=new int[len];
        int i ,j,k;
        i=j=k=0;
        while( i<nums1.length&&j<nums2.length){
            
            list[k++]=(nums1[i]>nums2[j] ? nums2[j++]:nums1[i++]); 
        }
        while(i<nums1.length)list[k++]=nums1[i++];    
        while(j<nums2.length)list[k++]=nums2[j++];
        //左邊 右邊
        double v=0;
        if((len&1)==1){
            v= 2*list[len/2];
        }else{
            v=list[len/2]+list[len/2-1];
        }
        return v/2;
    }
    
}