Median of Two Sorted Arrays
There are two sorted arrays A and B of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
1 public class Solution { 2 public double findMedianSortedArrays(int A[], int B[]) { 3 int alen = A.length; 4 int blen = B.length; 5 if((alen + blen) % 2 == 1) return findKthElement(A, B, 0, alen - 1, 0, blen - 1, (alen + blen + 1) / 2); 6 else return (double) (findKthElement(A, B, 0, alen - 1, 0, blen - 1, (alen + blen) / 2) + findKthElement(A, B, 0, alen - 1, 0, blen - 1, (alen + blen + 2) / 2)) / 2.0; 7 } 8 private int findKthElement(int arra[], int arrb[], int ast, int and, int bst, int bnd, int num){ 9 if(ast > and) return arrb[bst + num - 1]; 10 if(bst > bnd) return arra[ast + num - 1]; 11 int amid = (ast + and) / 2; 12 int bmid = (bst + bnd) / 2; 13 int len = amid - ast + 1 + bmid - bst + 1; 14 if(arra[amid] < arrb[bmid]){ 15 if(len <= num) return findKthElement(arra, arrb, amid + 1, and, bst, bnd, num - amid + ast - 1); 16 else return findKthElement(arra, arrb, ast, and, bst, bmid - 1, num); 17 }else{ 18 if(len <= num) return findKthElement(arra, arrb, ast, and, bmid + 1, bnd, num - bmid + bst - 1); 19 else return findKthElement(arra, arrb, ast, amid - 1, bst, bnd, num); 20 } 21 } 22 }
posted on 2014-07-31 09:44 Step-BY-Step 阅读(104) 评论(0) 编辑 收藏 举报