问题难度:Hard
源链接:https://leetcode.com/problems/median-of-two-sorted-arrays/description/
描述:
There are two sorted arrays nums1 and nums2 of size m and n respectively.
Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
思考与解读:
该问题要求找出给定的两个有序数组中的中位数,并且有一定的时间复杂度限制,不能把数组合并再排序这样来获得答案。
观察发现对于两个有序数组的情况和归并排序中merge的过程有一定相似性,可以通过merge的思路来处理这两个数组,
使用两个标记和一个计数器,两个标记在两个数组上递增,且只选择对应值相比较小的标记递增,计数器显示到总大小的二分之一停止,这样便获得了中位数且
对于数组中的每个数至多访问一次。
代码如下:
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int* result = new int[nums1.size()+nums2.size()+1];
int total = 1;
int p1 = 0;
int p2 = 0;
while((nums1.size()+nums2.size())/2+1 >= total){
if (p1 >= nums1.size() && p2 < nums2.size()) {
result[total] = nums2[p2];
p2++;
} else if (p1 < nums1.size() && p2 >= nums2.size()) {
result[total] = nums1[p1];
p1++;
} else {
if (nums1[p1] < nums2[p2]) {
result[total] = nums1[p1];
p1++;
} else {
result[total] = nums2[p2];
p2++;
}
}
total++;
}
double a = result[(nums1.size()+nums2.size())/2];
double b = result[(nums1.size()+nums2.size())/2+1];
if ((nums1.size()+nums2.size())%2 == 0) {
return (a+b)/2;
} else {
return b;
}
}
};