LeetCode第四题

weixin_57959966

已于 2023-04-13 19:16:48 修改

阅读量101

点赞数

分类专栏： Leetcode 文章标签： leetcode 算法

于 2023-04-10 16:55:34 首次发布

本文链接：https://blog.csdn.net/weixin_57959966/article/details/130064222

版权

Leetcode 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

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

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

算法的时间复杂度应该为 O(log (m+n)) 。
输入和输出的要求
方法：

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m = nums1.length, n = nums2.length; // m为nums1数组的长度,n为nums2数组的长度
        int total = m + n;   // m和n的总长度之和
        if(total %  2  == 1){
            // 如果总数为奇数
            return findKth(nums1,0,nums2,0,total / 2 + 1);  // 不理解
        }else{ // 如果总数为偶数
              return (findKth(nums1,0,nums2,0,total / 2) + findKth(nums1,0,nums2,0,total / 2 + 1)) / 2.0;  // 不理解
        }
    }
    // 下面都不理解
    private int findKth(int[] nums1,int i,int[] nums2,int j ,int k){
        if(i >= nums1.length) return nums2[j + k - 1]; // 如果nums1数组已经为空，则直接返回nums2中的第k个元素

        if(j >= nums2.length) return nums1[i + k - 1]; // 如果nums2数组已经为空，则直接返回nums1中的第k个元素

        if(k == 1) return Math.min(nums1[i],nums2[j]);//如果k等于1，则直接返回两个数组中第一个元素的最小值

        int mid1 = i + k / 2 - 1 < nums1.length ? nums1[i + k / 2 - 1] : Integer.MAX_VALUE; // 取得两个数组中第k/2个元素

        int mid2 = j + k / 2 - 1 < nums2.length ? nums2[j + k / 2 - 1] : Integer.MAX_VALUE; 

        if(mid1 < mid2){ // 如果nums1中第k/2个元素比nums2中第k/2个元素小
               return findKth(nums1,i+k / 2,nums2,j, k - k / 2);// 则可以删除nums1中前k/2个元素
        }else{ //如果nums1中第k/2个元素比nums2中第k/2个元素大
             return findKth(nums1,i, nums2, j + k / 2,k - k / 2);// 则可以删除nums2中前k/2个元素

        }
    }
}