您的位置:首页 > 编程语言 > C语言/C++

leetcode寻找两个有序数组的中位数C++版

2019-05-11 23:12 603 查看

leetcode寻找两个有序数组的中位数

题目描述:

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

nums1 = [1, 3]
nums2 = [2]
则中位数是 2.0

示例 2:

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

则中位数是 (2 + 3)/2 = 2.5

解决方案:

采用归并排序法:

  1. 判断nums1或nums2中是否为空,若其中一个为空,则直接判断另一个数组长度的奇偶性
  2. 如均不为空,则建立一个新数组temp,将nums1和nums2中数据依次比较大小并放入新数组temp中
  3. 如果其中一个数组比较完了,另一个数组还有剩余,则还要继续将这个数组放完
  4. 最后返回这个并归排序完的新数组,判断数组长度的奇偶性,得到结果

C++代码如下:

class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {

int m = nums1.size();
int n = nums2.size();
if (nums1.empty()) {
if (n%2 != 0)
return 1.0*nums2[n/2];
return (nums2[n/2]+nums2[n/2-1])/2.0;
}
if (nums2.empty()) {
if (m%2 != 0)
return 1.0*nums1[m/2];
return (nums1[m/2]+nums1[m/2-1])/2.0;
}
int i = 0;
int j = 0;
vector<int> temp;
while (i < m && j < n) {
if (nums1[i] <= nums2[j]) {
temp.push_back(nums1[i]);
i++;
} else {
temp.push_back(nums2[j]);
j++;
}
}
if (i < m) {
for (; i < m; i++)
temp.push_back(nums1[i]);
} else if (j < n) {
for (; j < n; j++)
temp.push_back(nums2[j]);
}
int len = temp.size();
if (len%2 != 0)
return 1.0*temp[len/2];
return (temp[len/2]+temp[len/2-1])/2.0;
}

};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: