两个数组求第k个数

遇到两个相似的题目，描述为：

1、两个数组求第K大数

题目描述

给定两个有序数组arr1和arr2，在给定一个整数k，返回两个数组的所有数中第K小的数。
例如：
arr1 = {1,2,3,4,5};
arr2 = {3,4,5};
K = 1;
因为1为所有数中最小的，所以返回1；

arr1 = {1,2,3};
arr2 = {3,4,5,6};
K = 4;
因为3为所有数中第4小的数，所以返回3；

要求：如果arr1的长度为N，arr2的长度为M，时间复杂度请达到O(log(min{M,N}))。

2、多数组中位数

题目描述

给定两个有序数组arr1和arr2，两个数组长度都为N，求两个数组中所有数的上中位数。
例如：
arr1 = {1,2,3,4};
arr2 = {3,4,5,6};
一共8个数则上中位数是第4个数，所以返回3。

arr1 = {0,1,2};
arr2 = {3,4,5};
一共6个数则上中位数是第3个数，所以返回2。

要求：时间复杂度O(logN)

看到题目，第一思路是利用归并的思路，用两个指针指向两个数组，从小到大遍历即可，但复杂度为O(m+n)。

看到是排序数组复杂度为log，也能想到应该是利用二分查找的思想，但是具体如何操作，却很久也没有想清楚。下面是参考了之前刷leetcode时记录的博客里的资料，重新写出来的。而中位数实际上是第(m+n)/2小的数，所以可以用同样的思路。

思路：首先假设数组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小值，所以我们可以将其抛弃。

下面是理解后，用java写出来的代码以及单元测试：

public class MultiArrayFindKth {

	public int getKth(int[] a, int i, int m, int[] b, int j, int n, int k) {
<span style="white-space:pre">		</span>if (m - i > n - j) {
<span style="white-space:pre">			</span>return getKth(b, j, n, a, i, m, k);
<span style="white-space:pre">		</span>}
<span style="white-space:pre">		</span>if (m == i) {
<span style="white-space:pre">			</span>return b[j + k - 1];
<span style="white-space:pre">		</span>}
<span style="white-space:pre">		</span>if (k == 1) {
<span style="white-space:pre">			</span>return Math.min(a[i], b[j]);
<span style="white-space:pre">		</span>}
<span style="white-space:pre">		</span>int pa = Math.min(k / 2, m - i);
<span style="white-space:pre">		</span>int pb = k - pa;
<span style="white-space:pre">		</span>if (a[i + pa - 1] == b[j + pb - 1]) {
<span style="white-space:pre">			</span>return a[i + pa - 1];
<span style="white-space:pre">		</span>} else if (a[i + pa - 1] < b[j + pb - 1]) {
<span style="white-space:pre">			</span>return getKth(a, i + pa, m, b, j, n, k - pa);
<span style="white-space:pre">		</span>} else {
<span style="white-space:pre">			</span>return getKth(a, i, m, b, j + pb, n, k - pb);
<span style="white-space:pre">		</span>}
<span style="white-space:pre">	</span>}
	
	@Test
	public void test1(){
		int a1[] = {1, 12, 15, 26, 38};
	    int a2[] = {2, 13, 17, 30, 45};
	    int resu = getKth(a1, 0, 5, a2, 0, 5, 5);
	    Assert.assertTrue(resu == 15);
	}
	
	@Test
	public void test2(){
		int a1[] = {1, 12, 15, 26, 38};
	    int a2[] = {2, 13, 17, 30, 45};
	    int resu = getKth(a1, 0, 5, a2, 0, 5, 1);
	    Assert.assertTrue(resu == 1);
	}
	
	@Test
	public void test3(){
		int a1[] = {1, 12, 15, 26, 38};
	    int a2[] = {2, 13, 17, 30, 45};
	    int resu = getKth(a1, 0, 5, a2, 0, 5, 9);
	    Assert.assertTrue(resu == 38);
	}
	
	@Test
	public void test4(){
		int a1[] = {1, 12, 15, 26, 38};
	    int a2[] = {2, 13, 17, 30, 45};
	    int resu = getKth(a1, 0, 5, a2, 0, 5, 10);
	    Assert.assertTrue(resu == 45);
	}
}