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

本文介绍了一种高效算法,用于在两个已排序数组中找到中位数,时间复杂度为O(m+n),空间复杂度为O(1)。

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

问题:

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

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


思路:我们首先判断中位数的位置,如果 m+n 为奇数,那么中位数的位置是 第 (m+n+1)/2。 如果 m + n 为偶数,那么我们要找出第 (m+n)/2 和 第(m+n)/2 + 1, 然后求平均。

我们在两个数组上都分配一个“指针”指到起始位(假定A数组的指针为pa, B数组的指针为pb),然后根据它们值的大小,指针不断的向前推进,直到总的被遍历的数的个数为(m+n+1)/2 (奇数情况)或者为(m+n)/2 (偶数情况)。

我们在移动指针对数组中的数进行比较的时候,我们要考虑指针是否已经操过数组的大小,然后才能进行比较,这是一个非常值得注意的地方。

public float median(int[] A, int[] B) { int m = A.length; int n = B.length; int pa = 0; int pb = 0; // to check whether the median is one or two. boolean isEven = (m + n)%2 == 0 ? true : false; float median = 0; int i = 0; if (isEven == false) { while (true) { i++; if (pa >= A.length) { if (i == (m+n+1)/2) { median = B[pb]; break; } pb++; } else if (pb >= B.length) { if (i == (m+n+1)/2) { median = A[pa]; break; } pa++; } else if (A[pa] <= B[pb]) { if (i == (m+n+1)/2) { median = A[pa]; break; } pa++; } else { if (i == (m+n+1)/2) { median = B[pb]; break; } pb++; } } } else { int median1 = 0; int median2 = 0; while (true) { i++; if (pa >= A.length) { if (i == (m+n)/2) { median1 = B[pb]; median2 = (pb < n-1 && B[pb+1] <= A[pa]) ? B[pb+1] : A[pa]; median = (float) (median1 + median2)/2; break; } pb++; } else if (pb >= B.length) { if (i == (m+n)/2) { median1 = A[pa]; median2 = (pa < m-1 && A[pa+1] <= B[pb]) ? A[pa+1] : B[pb]; median = (float) (median1 + median2)/2; break; } pa++; } else if (A[pa] <= B[pb]) { if (i == (m+n)/2) { median1 = A[pa]; median2 = (pa < m-1 && A[pa+1] <= B[pb]) ? A[pa+1] : B[pb]; median = (float) (median1 + median2)/2; break; } pa++; } else { if (i == (m+n)/2) { median1 = B[pb]; median2 = (pb < n-1 && B[pb+1] <= A[pa]) ? B[pb+1] : A[pa]; median = (float) (median1 + median2)/2; break; } pb++; } } } return median; }






评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值