[leetcode]Median of Two Sorted Arrays

Median of Two Sorted Arrays

Difficulty:Hard

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)).

Example 1:

nums1 = [1, 3]
nums2 = [2]

The median is 2.0

Example 2:

nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5

分析:
就是求两个排好序的数组的中值，如果两个数组的数字个数和是奇数，则中值就是两个数组排序的中间值，如果是偶数，就是两个数组排序的中间两个值的均值。

因为对时间复杂度有要求，要求达到O(log (m+n))， 所以不能对所有的数进行排序后再选取中间值，应该要提前计算好中间值是哪个。

在读取数组时，会出现一个数组已经遍历完了，还没有找到中值的情况，需要注意这种情况，不要越界读取数组。

我的实现方法是先求两个数组长度和的一半并取整（假设为m），找到两个数组排序时第m位以及第m+1位的数值，如果两个数组长度和是奇数，则第m+1位的数值就是中值，如果两个数组长度
4000
和是偶数，则第m位以及第m+1位数值的和的均值就是中值。 double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int quantity = nums1.size() + nums2.size();//两个数组长度和
float sit = (float)quantity / 2;//长度和的一半
int i_sit = (int)sit;//长度和的一半再取整
int a=0,b=0,k=0;
int median = -1;//保存当前中值数值
int next = -1;//保存当前中值的下一位数值

while (k < i_sit+1){//在k达到当前中值的下一位之前一直比对
if (a == nums1.size()){//第一个数组已经遍历完的情况
median = next;
next = nums2.at(b);
b++;
k++;
continue;
}
if (b == nums2.size()){//第而个数组已经遍历完的情况
median = next;
next = nums1.at(a);
a++;
k++;
continue;
}
if (nums1.at(a) < nums2.at(b)){
median = next;
next = nums1.at(a);
a++;
k++;
}
else{
median = next;
next = nums2.at(b);
b++;
k++;
}
}

if (sit - (int)sit == 0){//数组长度和为偶数
return (double)(median + next) / 2;
}
else{//数组长度和为奇数
return (double)next;
}
}