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

提示:

  • List item
  • nums1.length == m
  • nums2.length == n
  • 0 <= m <= 1000
  • 0 <= n <= 1000
  • 1 <= m + n <= 2000
  • -106 <= nums1[i], nums2[i] <= 106

进阶:你能设计一个时间复杂度为 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)) 较难,思路为二分法
目前未完成, 后面补充
上一篇:瓶子君前端算法集训营第一期开营啦,免费哟


下一篇:初级算法-06-两个数组的交集 II