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

说明

给定两个大小分别为 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

提示：

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

来源：力扣（LeetCode） 链接：leetcode.cn/problems/me…

题解

解法

    public static void main(String[] args) {
        System.out.println(demo(new int[]{1,2},new int[]{3,4}));
    }

    public static  double demo(int[] nums1, int[] nums2){
        int number = nums1.length + nums2.length;
        int[] sums = new int[(number) / 2 + 1];
        int num1 = 0;
        int num2 = 0;
        for (int i = 0; i < sums.length; i++) {
            if(num1 == nums1.length){
                sums[i] = nums2[num2];
                num2++;
            }else if(num2 == nums2.length){
                sums[i] = nums1[num1];
                num1++;
            }else{
                int n1 = nums1[num1];
                int n2 = nums2[num2];
                if(n1 < n2){
                    sums[i] = n1;
                    num1++;
                }else{
                    sums[i] = n2;
                    num2++;
                }
            }
        }
        return number % 2 == 1? sums[sums.length - 1] : (double)(sums[sums.length - 1] + sums[sums.length - 2]) / 2;
    }