其他分享
首页 > 其他分享> > leetcode 004.找两个正序数组的中位数

leetcode 004.找两个正序数组的中位数

作者:互联网

题目:

给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
https://leetcode-cn.com/problems/median-of-two-sorted-arrays/

示例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

示例3:

输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000

示例4:

输入:nums1 = [], nums2 = [1]
输出:1.00000

示例5:

输入:nums1 = [2], nums2 = []
输出:2.00000

提示:

进阶:你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗?

解题思路:

简单解题:
两数组长度和为奇数  中位数为 中间数的值
两数组长度和为偶数  中位数为 中间两数的平均值
1 求中间数的值:
1)两个数组分别有一个指向0的指针,比较两个指针指向的值的大小,取较小的值,然后较小的值对应的指针向右移动
2)边界:如果指针已经移动超过数组界限,直接取另一个数组的值
2 根据长度和求中位数:
1) 奇数 直接返回中间数的值
2) 偶数 获取中间数的下一位较小值,返回 他和中间数的平均值 

JAVA:

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int idx1 = 0 , idx2 = 0;
        double left=0.0 ;
        for(int i=0; i< (nums1.length + nums2.length + 1)/2 ; i++ ){

            if(idx1 == nums1.length){
                left = nums2[idx2++];
                continue;
            }

            if(idx2 == nums2.length){
                left = nums1[idx1++];
                continue;
            }
 
            left = nums1[idx1] < nums2[idx2] ? nums1[idx1++] : nums2[idx2++];
        }

        if((nums1.length + nums2.length )%2 == 1){
            return left;
        }

        double right  = 0.0;

        if(idx1 == nums1.length){
            right = nums2[idx2++];
        } else if(idx2 == nums2.length){
            right = nums1[idx1++];
        } else { 
            right = nums1[idx1] < nums2[idx2] ? nums1[idx1++] : nums2[idx2++];
        }

        return (left+ right)/2.0;
    }
}
进阶: 时间复杂度为 O(log (m+n)) 较难,思路为二分法
目前未完成, 后面补充

标签:正序,++,nums1,length,004,idx2,idx1,leetcode,nums2
来源: https://blog.csdn.net/u012926411/article/details/116865810