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

最新推荐文章于 2019-04-15 22:06:38 发布

beiyeqingteng

最新推荐文章于 2019-04-15 22:06:38 发布

阅读量5k

点赞数

CC 4.0 BY-SA版权

分类专栏：算法 (Algorithm) 文章标签： float null string class

本文链接：https://blog.csdn.net/beiyeqingteng/article/details/6992685

算法 (Algorithm) 专栏收录该内容

113 篇文章

订阅专栏

本文介绍了一种求解两个已排序数组中位数的方法，时间复杂度为O(m+n)，空间复杂度为O(1)。通过遍历两个数组找到第(m+n+1)/2个元素，并根据数组总长度的奇偶性返回中位数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题：

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