leetcode腾讯精选-004

最新推荐文章于 2022-03-17 22:14:50 发布

goodyang1

最新推荐文章于 2022-03-17 22:14:50 发布

阅读量103

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/goodyang1/article/details/90286107

博客给出LeetCode上两有序数组中位数问题的题目链接及代码实现，代码使用C++编写，通过递归函数求解。博主表示自己写的代码存在问题，主要是边界条件考虑不足，答案参照了网上内容。

题目：https://leetcode-cn.com/problems/median-of-two-sorted-arrays/
代码：class Solution {
public:
double findMedianSortedArrays(vector& nums1, vector& nums2) {
int m = nums1.size(), n = nums2.size(), left = (m + n + 1) / 2, right = (m + n + 2) / 2;
return (findKth(nums1, 0, nums2, 0, left) + findKth(nums1, 0, nums2, 0, right)) / 2.0;
}
int findKth(vector& nums1, int i, vector& nums2, int j, int k) {
if (i >= nums1.size()) return nums2[j + k - 1];
if (j >= nums2.size()) return nums1[i + k - 1];
if (k == 1) return min(nums1[i], nums2[j]);
int midVal1 = (i + k / 2 - 1 < nums1.size()) ? nums1[i + k / 2 - 1] : INT_MAX;
int midVal2 = (j + k / 2 - 1 < nums2.size()) ? nums2[j + k / 2 - 1] : INT_MAX;
if (midVal1 < midVal2) {
return findKth(nums1, i + k / 2, nums2, j, k - k / 2);
} else {
return findKth(nums1, i, nums2, j + k / 2, k - k / 2);
}
}
};