LeetCode. 寻找两个有序数组的中位数(二分技巧 + 边界处理问题)

5 篇文章 0 订阅
4 篇文章 0 订阅
核心讲解可参考:https://blog.csdn.net/hit1110310422/article/details/80865539
i 值用于切割nums1,j 用于切割num2
对 nums1数组 进行二分查找,搜寻到恰好符合要求的 i 值
要求: B[ j − 1 ] ≤ A[ i ] 且 A[ i - 1 ] ≤ B[ j ]
前提需要满足 :
1、Length(nums1) <= Length(nums2)
2、切割后左边累计数量 等于 右边累计数量: j = ( m + n + 1 ) / 2 - i (推导出的公式,当二分求出 i 值后,直接使用此公式求出 j 值即可满足要求)
#include <vector>
#include <algorithm>
using namespace std;
class Solution {
public:
	//num1.size() <= nums.size()
	double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
		int m = nums1.size();
		int n = nums2.size();
		if (m > n)
			return findMedianSortedArrays(nums2, nums1);
		int iBegin = 0, iEnd = m, halfLen = (m + n + 1) / 2;
		while (iBegin <= iEnd) {
			//对 num1 进行二分搜索
			int i_1_mid = (iBegin + iEnd) / 2;
			int j_2_mid = halfLen - i_1_mid;
			//要注意避免i_1_mid  = iEnd 或者 i_1_mid = iBegin 的情况
			if (i_1_mid < iEnd && nums2[j_2_mid - 1] > nums1[i_1_mid])
				iBegin = i_1_mid + 1; // i 太小
			else if (i_1_mid > iBegin && nums1[i_1_mid - 1] > nums2[j_2_mid])
				iEnd = i_1_mid - 1; // i 太大
			else { // i 命中目标,分奇数和偶数.
				//奇数 情况
				int maxLeft = 0;
				if (i_1_mid == 0)
					maxLeft = nums2[j_2_mid - 1];
				else if (j_2_mid == 0)
					maxLeft = nums1[i_1_mid - 1];
				else
					maxLeft = max(nums1[i_1_mid - 1], nums2[j_2_mid - 1]);
				if ((m + n) % 2 == 1)
					return maxLeft;
				//偶数 情况
				int minRight = 0;
				if (i_1_mid == m)
					minRight = nums2[j_2_mid];
				else if (j_2_mid == n)
					minRight = nums1[i_1_mid];
				else
					minRight = min(nums2[j_2_mid], nums1[i_1_mid]);
				return (static_cast<double>(maxLeft) + minRight) / 2.0f;
			}
		}
		return 0.0f;
	}
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值