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)).
思路:
二分查找变形
我的代码:
public class Solution { public double findMedianSortedArrays(int A[], int B[]) { int lenA = A.length; int lenB = B.length; if((lenA+lenB)%2 == 1) return (double)findKth(A,B,(lenA+lenB)/2+1); else return (findKth(A,B,(lenA+lenB)/2+1) + findKth(A,B,(lenA+lenB)/2))/(double)2; } public int findKth(int[] a, int[] b, int k) { int lenA = a.length; int lenB = b.length; if(lenA > lenB) return findKth(b, a, k); if(lenA == 0) return b[k-1]; if(k == 1) return Math.min(a[0],b[0]); int pa = Math.min(k/2,lenA); int pb = k - pa; if(a[pa-1] == b[pb-1]) return a[pa-1]; else if(a[pa-1] < b[pb-1]) return findKth(Arrays.copyOfRange(a, pa, lenA), b, k - pa); else return findKth(a, Arrays.copyOfRange(b, pb, lenB), k - pb); } }
学习之处:
- 今天下午状态不好,想了一会没有思路,就直接看答案了,之前肯定是用二分查找的问题,没想到二分查找也等价于二分排除嘛!这一点没有意识到。
- 原题被转换成经典的找两个sorted的数组中 K大的数,不断的去排除掉不符合的部分
- 首先假设数组A和B的元素个数都大于k/2,我们比较A[k/2-1]和B[k/2-1]两个元素,这两个元素分别表示A的第k /2小的元素和B的第k/2小的元素。这两个元素比较共有三种情况:>、<和=。如果A[k/2-1]<B[k/2-1],这表示 A[0]到A[k/2-1]的元素都在A和B合并之后的前k小的元素中。换句话说,A[k/2-1]不可能大于两数组合并之后的第k小值,所以我们可以将 其抛弃,同理A[k/2-1]>B[k/2-1]。当A[k/2-1]=B[k/2-1]的时候,A[k/2 - 1]即为第K个大的数