Lcode算法29：寻找两个正序数组的中位数

似水_逆行

于 2024-09-05 10:33:08 发布

阅读量72

点赞数 1

分类专栏：面试算法文章标签：算法

本文链接：https://blog.csdn.net/qq_43462685/article/details/141924632

版权

面试算法专栏收录该内容

35 篇文章 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

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
            int length = nums1.length + nums2.length;
            int[] result = new int[length];
            int index=0;
            for(int i=0,j=0;i<nums1.length || j<nums2.length;){
                if(i>=nums1.length){
                    result[index] = nums2[j];
                    j++;
                    index++;
                    continue;
                }
                 if(j>=nums2.length){
                    result[index] = nums1[i];
                    i++;
                    index++;
                    continue;
                }
                if(nums1[i]<=nums2[j]){
                    result[index] = nums1[i];
                    i++;
                }else{
                    result[index] = nums2[j];
                    j++;
                }
                index++;
            }
            Double val = 0.0d;
            if(length%2 == 0){
                val = (result[length/2] + result[(length/2) -1] + 0.0d)/2;
            }else{
                val = result[(length/2)] + 0.0d;
            }
            return val;
    }
}