寻找两个有序数组的中位数
![标题](https://i-blog.csdnimg.cn/blog_migrate/3abaec24114fea63cc4190f9bd295051.png)
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);
}
};
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/f3935546b5f7d7bb2c5a447e5728eee7.png)