Leetcode 第四题实现两排序数组中位数 与C++vector应用

1.题目

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。

请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。

示例 1:

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

中位数是 2.0

示例 2:

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

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

2.思路

定义第三个向量,对两个向量从头开始比较,较小的用尾插发插入第三个向量末尾,最后判断向量的大小.根据奇偶性取出中位数:若为奇数则中位数为下标等于向量大小的一半;若为偶数则位下标一半与妻哪一个和的一半.

向量排序插入实现:以最小向量大小为循环,采用三目运算符实现较小的数用.pash_back()插入末尾;跳出循环后,把剩余的数一次插入向量即可;

代码如下:

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        vector<int>num; //定义第三个向量
        int i,j;
        for(i=0,j=0;i<nums1.size()&&j<nums2.size();)
        {
	        nums1[i]<nums2[j] ? num.push_back(nums1[i++]) : num.push_back(nums2[j++]);  //三目运算,较小者插入向量
        }
        if(i>=nums1.size()){for(;j<nums2.size();j++) num.push_back(nums2[j]);}  //若nums1存在未插入数据,依次插入
        else if(j>=nums2.size()){for(;i<nums1.size();i++) num.push_back(nums1[i]);}//若nums2存在未插入数据,依次插入
        i=num.size()/2; //获得向量的中间下标
        if(num.size()%2==0) //判断向量的奇偶性
            return (num[i]+num[i-1])/2.0;//偶数
        else
            return num[i];  //奇数
        return 0;  
    }
};

时间复杂度:存在两个for循环:第一个与第二个一共运行i+j次故时间复杂度为O(m+n);

3.感想

在运用向量时对向量的定义初始化做了更深刻了解;注意向量越界后并为定义故其值不确定;因此不能做向量越界比较,否则容易出错.

向量定义时,最开始定义只能用.pash_back()初始化,不能用下标num[0]写法.

三目运算符简化代码写法,看起来比if简答整洁.

如有更好写法,下面留言讨论.