1.题意
输入为两个数字数组,找出其中的中位数
要求时间复杂度为log(m+n)
2.思路
先把两个数组放入一个容器中,而后找中位数,
如果容器size为奇数,则是中位数本身,偶数,则是中间两数字的平均值
使用堆或者快速排序还要重新造轮子,搜到一个很实用的stl方法
找数组或容器中的第n大的方法
std::nth_element(RandomIt first, RandomIt nth, RandomIt last)
first,last为第一个和最后一个迭代器,nth为要寻找的第n大的元素
排序后
(1)第n大的数字在第n的位置上
(2)左边的数字都比第n位置数字小,右边的数字都比第n位的数字大
3.反思
奇数操作有错误
v1.begin()+(v1.size()+1)/2
v1.begin()本就是指向第一个的迭代器
要做的是迭代到第n个
正确示范
v1.begin()+(v1.size())/2
4.其余收获
思路醉清晰的一次,也是自己搜索最简明的一次,半小时
顺便捋了捋快速排序的机制
另外一提vector的size就是元素的个数
capacity是已经分配的存储容量
不像hard题目
5.代码
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
vector<int> v1;
v1.insert(v1.end(),nums1.begin(),nums1.end());
v1.insert(v1.end(),nums2.begin(),nums2.end());
if(v1.size()%2!=0)
{
nth_element(v1.begin(), v1.begin()+(v1.size())/2, v1.end());
double result = *(v1.begin()+(v1.size())/2);
return result;
}
else
{
nth_element(v1.begin(), v1.begin()+v1.size()/2, v1.end());
nth_element(v1.begin(), v1.begin()+(v1.size()-1)/2, v1.end());
double a = *(v1.begin()+ v1.size()/2);
double b = *(v1.begin()+(v1.size()-1)/2);
double result = (a+b)/2;
return result;
}
}
};