题目描述
给定两个大小为 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
题目分析
该题主要是对时间复杂度的要求,实现起来要充分利用题中的已经知道两个数组有序这一条件,进行解答。设置两个指针i,j;分别指向数组1和数组2的首元素,比较他们的大小,按照从小到大的顺序加入到新的数组中。因为两个数组有序,因此需要做的工作的就是从开始的i,j分别指向的第一个元素开始,将小的加入到新数组中,并将指针向后移动,不断的比较i,j指向的元素,那个指针指向的 元素小就将其加入到新数组中,对应的指针向后移动。实际上就是在对两个数组进行合并排序。
注意这里的结束条件,就是并不需要将两个数组完全的合并起来,因为只需要我们合并到能确定中位数即可。
// ConsoleApplication4.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//
#include<iostream>
#include<string>
#include<vector>
using namespace std;
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2);
int main() {
vector<int> n1 = {2,3};
vector<int> n2 = {};
double m = findMedianSortedArrays(n1, n2);
cout << m;
return 0;
}
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int Middle_index = (nums1.size() + nums2.size()) / 2;
vector<int> com;
if (nums1.size() == 0) {
return nums2.size() % 2 == 0?(double)(nums2[Middle_index - 1]+ nums2[Middle_index])/2: nums2[Middle_index];
}
if (nums2.size() == 0) {
return nums1.size() % 2 == 0?(double)(nums1[Middle_index - 1] + nums1[Middle_index])/2: nums1[Middle_index];
}
for (int i = 0, j = 0; com.size() <= Middle_index;) {
while (i < nums1.size()&& j < nums2.size())
{
if (nums1[i] <= nums2[j]) {
com.push_back(nums1[i]);
i++;
}
else
{
com.push_back(nums2[j]);
j++;
}
}
if (i == nums1.size()&&j!= nums2.size()) {
com.push_back(nums2[j]);
j++;
}
if (j == nums2.size()&&i!=nums1.size()) {
com.push_back(nums1[i]);
i++;
}
}
double median;
if ((nums1.size() + nums2.size()) % 2 == 0) {
median =(double)(com[Middle_index - 1] + com[Middle_index])/2;
return median;
}
else
{
median = com[Middle_index];
return median;
}
}
执行耗费时间还可以。