问题:
给两个已经排好序的数组,一个长度为 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