4. 寻找两个有序数组的中位数

Posted 2022-09-04 doona-jazen

题目：

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。

示例 1:
nums1 = [1, 3]
nums2 = [2]
则中位数是 2.0

示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
则中位数是 (2 + 3)/2 = 2.5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays/

分析：

首先想到的是将数组所有元素放在同一个数组中，然后进行排序，排序之后取中位数就简单了，按照此思路得到以下代码：

class Solution 
    public static double findMedianSortedArrays(int[] nums1, int[] nums2) 
        int[] nums = join(nums1,nums2);
        double num = midMethod(nums);
        return num;
    
    /**
     * 将两个数组元素组合到一个数组中，返回结果有序
     */
    public static int[] join(int a[],int b[]) 
        List<Integer> testList = new ArrayList<Integer>();
        //遍历添加给入数组元素
        for (int j = 0; j < a.length; j++) 
            testList.add(a[j]);
        
        for (int k = 0; k < b.length; k++) 
            testList.add(b[k]);
        
        int c[] = new int[testList.size()];
        for (int i = 0; i < alist.size(); i++) 
            c[i] = alist.get(i);
        
        Arrays.sort(c);
        return c;
    
    /**
     * 得到中位数
     */
    public static double midMethod(int[] nums) 
        double mid = nums.length / 2.0 ;
        double num = 0;
        if (mid == 0) 
            //若给定的两个数组都为空，不做处理
         else if (Math.floor(mid) != mid) 
            num = nums[(int)mid];
         else 
            num = (nums[(int)mid-1] + nums[(int)mid]) / 2.0;
        
        return num;
    

View Code