leecode 4st Median of Two Sorted Arrays

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;  
        }
    }
};

你可能感兴趣的:(leetcode,Practice)