4. 寻找两个正序数组的中位数 xp_xht123

最新推荐文章于 2024-11-09 23:37:51 发布

xp_xht123

最新推荐文章于 2024-11-09 23:37:51 发布

阅读量122

点赞数

分类专栏： leetcode 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/xp_xht123/article/details/126553882

版权

leetcode 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。

算法的时间复杂度应该为 O(log (m+n)) 。

示例 1：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2'

示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

解题思路1：类似于快排+归并的操作一样，双指针，遍历一遍两个数组，复杂度应该为 $O(m + n)$

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) 
    {
        int m = nums1.size();
        int n = nums2.size();

        double res = 0;
        int x = m + n;
        int i = 0 , j = 0;
        x >>= 1;
        int t[2010];
        int cnt = 0;
        while(i < nums1.size() || j < nums2.size())
        {
            if(i != nums1.size() && j != nums2.size())
            {
                if(nums1[i] < nums2[j]) t[cnt ++] = nums1[i ++];
                else t[cnt ++] = nums2[j ++];
            }
            if(i != nums1.size() && j == nums2.size()) t[cnt ++] = nums1[i ++];
            if(i == nums1.size() && j != nums2.size()) t[cnt ++] = nums2[j ++];
        }
        if((m + n) % 2 == 0) res = (double)(t[x] + t[x - 1]) / 2;
        else res = (double)t[x];
        return res;
    }
};