找出两个已经排好序的数组的中位数

来源:互联网 发布:433m模块stc单片机解码 编辑:程序博客网 时间:2024/05/16 15:44

问题:

给两个已经排好序的数组,一个长度为 m (m >= 1), 一个长度为 n (n >= 1),找出这两个数组的中位数。时间复杂度要求为 O(m+n), 空间复杂度为 O(1)。

其实这个问题本身来讲是不难的,关键的关键,是对一些边界条件的处理。

思路:

我们首先不管两个数组里含有多少个数字,而是先得到第 (m+n+1)/2个数字的值。 如果 m + n为奇数,我们返回第 (m+n+1)/2,如果m + n为偶数,那么我们找出第 (m+n+2)/2个数字, 然后求平均。

public class Med {public static float getMedian(int[] a, int[] b) {if (a == null || b == null || (a.length + b.length) == 0) return Float.MIN_VALUE;int pa = 0;int pb = 0;float median = 0;while (pa + pb != (a.length + b.length + 1) / 2) {int Ai = (pa == a.length) ? Integer.MAX_VALUE : a[pa];          int Bj = (pb == b.length) ? Integer.MAX_VALUE : b[pb];         if (Ai < Bj) {        median = a[pa];        pa++;        } else {        median = b[pb];        pb++;        }}if ((a.length + b.length) % 2 == 1) {return median;} else {int Ai = (pa == a.length) ? Integer.MAX_VALUE : a[pa];          int Bj = (pb == b.length) ? Integer.MAX_VALUE : b[pb]; int median2 = (Ai < Bj) ? Ai : Bj;return (float)(median + median2) / 2;}}public static void main(String[] args) {                  int[] b = {1, 2};          int[] a = {5, 6};  System.out.println(getMedian(a,b));    }}

转载请注明出处:http://blog.csdn.net/beiyeqingteng