1. 程式人生 > 其它 >二分5.尋找兩個排序陣列的中位數

二分5.尋找兩個排序陣列的中位數

技術標籤:演算法-陣列資料結構演算法javaleetcode

首先如果使用歸併陣列找到中間數,則空間複雜度為O(n),時間複雜度為O(n),如果利用一個指標來達到找中位數的過程,模擬歸併但不歸併,那時間複雜度仍為O(n)
由於題中說了是有序的,所以可以嘗試二分,這個題本質就是找到陣列中第k小的數,k為(m+n)/2或者是(m+n)/2和(m+n)/2+1的平均值。
先比較A[k/2-1]和B[k/2-1],若A[k/2-1]<B[k/2-1],則比A[k/2-1]小的數肯定不可能成為k,則直接排除掉A[k/2-1]以及比它小的,可以排除掉k/2的數字,並根據排除掉的數字來減小k的值直到k為0.排除就利用index的增大來表現一個數組的開始在哪裡,因為不會排除後面的,所以不用設定後一個。

如果k=1,則返回兩個陣列的首元素的最小值,如果一個數組為空,則直接可以返回另一個數組中第k個數字,如果越界,則選擇最後一個元素,記得根據實際排除的值來減小k。

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int length1=nums1.length;
        int length2=nums2.length;
        int length=length1+length2;
        if((length&1) ==
1){ int k=length/2+1; return core(nums1,nums2,k); }else{ int k1=length/2; int k2=length/2+1; return (core(nums1,nums2,k1)+core(nums1,nums2,k2))/2; } } public double core(int[] nums1,int[] nums2,int k){ int i1=0; int i2=
0; while(true){ if(i1==nums1.length){ return nums2[i2+k-1]; } if(i2==nums2.length){ return nums1[i1+k-1]; } if(k==1){ return (double)Math.min(nums1[i1],nums2[i2]); } int half=k/2; int newIndex1=Math.min(half+i1,nums1.length)-1; int newIndex2=Math.min(half+i2,nums2.length)-1; if(nums1[newIndex1]<=nums2[newIndex2]){ k-=(newIndex1-i1+1); i1=newIndex1+1; }else{ k-=(newIndex2-i2+1); i2=newIndex2+1; } } } }