序言
记录自己的算法升级打怪之路,持续更新
Leetcode系列
Leetcode P4.寻找两个正序数组的中位数
题目:
思路:如果没有时间复杂度,可以直接合并然后算中位数,但是设定了时间复杂度,由时间复杂度可知用二分来做。但是此题二分难度极大,极难理解,参考题解力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
//任何长度的数组的中位数 可以用(length+1)/2 + (length+2)/2除2得到
int left = (nums1.size() + nums2.size() + 1) / 2;
int right = (nums1.size() + nums2.size() + 2) / 2;
return (findmid(nums1, nums2, 0, 0, left) + findmid(nums1, nums2, 0, 0, right)) / 2;
}
double findmid(vector<int>& arr1, vector<int>& arr2, int i, int j, int k) {
//如果i超过了数组num1的长度,剩下的k直接在num2里找
if (i >= arr1.size()) {
return arr2[j + k - 1];
}
if (j >= arr2.size()) {
return arr1[i + k - 1];
}
//如果k只剩下一个了,说明这就是答案,选择两个数组中最小的那个
if (k == 1) { return min(arr1[i], arr2[j]); }
int mid1 = 0;
int mid2 = 0;
if (i + k / 2 - 1 < arr1.size()) {
mid1 = arr1[i + k / 2 - 1];
}
else {
mid1 = INT_MAX;
}
if (j + k / 2 - 1 < arr2.size()) {
mid2 = arr2[j + k / 2 - 1];
}
else {
mid2 = INT_MAX;
}
if (mid1 < mid2) {
return findmid(arr1, arr2, i + k / 2, j, k - k / 2);
}
else {
return findmid(arr1, arr2, i, j + k / 2, k - k / 2);
}
}
};
提交结果: