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

幻想小蜗

于 2023-04-09 16:00:07 发布

阅读量129

点赞数 1

CC 4.0 BY-SA版权

文章标签： leetcode 算法 c++ 数据结构

本文链接：https://blog.youkuaiyun.com/Kerwinstudy/article/details/130043377

该代码实现了一个方法，用于在限定复杂度的条件下合并两个已排序的整数数组，并计算合并后的数组中位数。通过while循环比较并合并数组元素，确保最终数组依然有序。最后根据数组长度确定中位数的值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

解题思路

由于限定了复杂度，只能在一次循环中获取合并的序列，所以使用while循环，设定两个索引，将
较小值插入新的数组中，在通过简单的计算获取新数组的中位数。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        
        int num1,num2;
        int index1=0;
        int index2=0;
        vector<int> v;
        int m=nums1.size();
        int n=nums2.size();
        while(index1<m&&index2<n){
            if(nums1[index1]<=nums2[index2]){
                v.push_back(nums1[index1]);
                index1++;
            }else {
                v.push_back(nums2[index2]);
                index2++;
            }
        }
        if(index1==m){
            while(index2<n){
                v.push_back(nums2[index2]);
                index2++;
            }
        }else if(index2==n){
            while(index1<m){
                v.push_back(nums1[index1]);
                index1++;
            }
        }
        int length=v.size();
        double mid=0;
        if(length%2==0){
            mid=(double)(v[length/2-1]+v[length/2])/2;
        }else{
            mid=v[(length+1)/2-1];
        }
        return mid;
    }
};