题目:求两个已排序数组合并之后形成的排序数组的中位数。
思路一:利用归并排序的思想分别从两个数组中取数据,直到取到第(m+n)/2数据为止(m、n分别为两个数组A、B的长度)。
思路二:将原问题转变成一个寻找两个已排序数组中前k小数的问题。在此的k等于(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小值,所以我们可以将其抛弃。这样我们就对被抛弃一部分之后的两个排序数组求前(k-被抛弃数组的长度)小数的问题—利用递归。
- 当A[k/2-1]>B[k/2-1]时存在类似的结论。
- 当A[k/2-1]=B[k/2-1]时,我们已经找到了第k小的数,也即这个相等的元素,我们将其记为m。由于在A和B中分别有k/2-1个元素小于m,所以m即是第k小的数。(这里可能有人会有疑问,如果k为奇数,则m不是中位数。这里是进行了理想化考虑,在实际代码中略有不同,是先求k/2,然后利用k-k/2获得另一个数。
- 另外一种情况是当数组A、B中有一个数组的长度小于k/2的时候(例如是数组A),此时,我们应该取数组A的最后一个值与数组B的第k-m个数进行上面的比较操作。
此外我们还需要考虑几个边界条件:
如果A或者B为空,则直接返回B[k-1]或者A[k-1];
如果k为1,我们只需要返回A[0]和B[0]中的较小值;
如果A[k/2-1]=B[k/2-1],返回其中一个;