程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> [LeetCode]Median of Two Sorted Arrays

[LeetCode]Median of Two Sorted Arrays

編輯:C++入門知識

[cpp] 
class Solution { 
//more detail refer to: http://fisherlei.blogspot.com/2012/12/leetcode-median-of-two-sorted-arrays.html  
//using the method of getting the kth number in the two sorted array to solve the median problem  
//divide-and-conquer  
//very clean and concise  
public: 
    double findMedianSortedArrays(int A[], int m, int B[], int n) {   
        if((n+m)%2 ==0)   
        {   
            return (GetMedian(A,m,B,n, (m+n)/2) + GetMedian(A,m,B,n, (m+n)/2+1))/2.0;   
        }   
        else   
            return GetMedian(A,m,B,n, (m+n)/2+1);         
    }   
    int GetMedian(int a[], int n, int b[], int m, int k)//get the kth number in the two sorted array    
    {   
        //assert(a && b);     
        if (n <= 0) return b[k-1];   
        if (m <= 0) return a[k-1];   
        if (k <= 1) return min(a[0], b[0]);    
        //a: section1 section2  
        //b: section3 section4  
        if (b[m/2] >= a[n/2])   
        {   
            if ((n/2 + 1 + m/2) >= k)   
                return GetMedian(a, n, b, m/2, k);//abort section 4    
            else   
                return GetMedian(a + n/2 + 1, n - (n/2 + 1), b, m, k - (n/2 + 1)); //abort section 1   
        }   
        else   
        {   
            if ((m/2 + 1 + n/2) >= k)   
                return GetMedian( a, n/2,b, m, k);//abort section 2  
            else   
                return GetMedian( a, n, b + m/2 + 1, m - (m/2 + 1),k - (m/2 + 1));//abort section 3  
        }   
    }   
}; 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved