leetcode刷题--(4)--寻找两个正序数组的中位数

一、题目描述

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、完整代码

class Solution {

    public static void main(String[] args) {
        int nums1[] = {0, 0, 0, 0, 0};
        int nums2[] = {-1, 0, 0, 0, 0, 0, 1};

        double median_num = findMedianSortedArrays(nums1, nums2);
        System.out.println("Median Number is ：" + median_num);

    }

    public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
        double sum = 0.00000;
        int num_len1 = nums1.length;
        int num_len2 = nums2.length;
        if (num_len1 > 1000 || num_len2 > 1000) {
            return sum;
        }
        int merge_num[] = new int[num_len1 + num_len2];

        int num1_index = 0;
        int num2_index = 0;
        int merge_index = 0;
        while (num1_index < num_len1 && num2_index < num_len2) {

            if (nums1[num1_index] < nums2[num2_index]) {
                merge_num[merge_index++] = nums1[num1_index++];
            } else if (nums1[num1_index] > nums2[num2_index]) {
                merge_num[merge_index++] = nums2[num2_index++];
            } else {
                merge_num[merge_index++] = nums1[num1_index++];
            }

        }
        if (num1_index == num_len1) {
            while (num2_index < num_len2)
                merge_num[merge_index++] = nums2[num2_index++];
        } else {
            while (num1_index < num_len1)
                merge_num[merge_index++] = nums1[num1_index++];
        }

        if (merge_num.length % 2 == 0) {
            sum = (merge_num[merge_num.length / 2 - 1] + merge_num[merge_num.length / 2]) / 2.00000;
        } else {
            sum = merge_num[(merge_num.length - 1) / 2];
        }
        return sum;
    }
}