LeetCode 4 寻找两个有序数组的中位数

寻找两个有序数组的中位数

class Solution {
public:
    //仿照别人自己理解写出来的d
    double find_k(int a_begin, int b_begin, vector<int>& nums1, vector<int>& nums2, int k)
{
	if (a_begin >= nums1.size())
		return nums2[b_begin + k - 1];
	if (b_begin >= nums2.size())
		return nums1[a_begin + k - 1];
	if (k == 1)
		return min(nums1[a_begin], nums2[b_begin]);
	int nums1_max = INT_MAX;
	int nums2_max = INT_MAX;
	if (a_begin + k / 2 - 1 < nums1.size())
		nums1_max = nums1[a_begin + k / 2 - 1];
	if (b_begin + k / 2 - 1< nums2.size())
		nums2_max = nums2[b_begin + k / 2 - 1];
	if (nums1_max < nums2_max)
		return find_k(a_begin + k / 2, b_begin, nums1, nums2, k - k / 2);
	else
		return find_k(a_begin, b_begin + k / 2, nums1, nums2, k - k / 2);
}
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
	int len1 = nums1.size();
	int len2 = nums2.size();
	int k = len1 + len2;
	if (len1 == 0 && len2 == 0)
		return 0;
	if (len1 == 0)
	{
		if (len2 % 2 == 0)
			return 0.5*(nums2[k / 2-1]+nums2[k/2]);
		else
			return nums2[k / 2];
	}
	if (len2 == 0)
	{
		if (len1 % 2 == 0)
			return 0.5*(nums1[k / 2-1] + nums1[k / 2]);
		else
			return nums1[k / 2];
	}
	if (k % 2 == 0)
		return 0.5*(find_k(0, 0, nums1, nums2, k / 2) + find_k(0, 0, nums1, nums2, k / 2 + 1));
	else
		return find_k(0, 0, nums1, nums2, k / 2+1);
}
};