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

最新推荐文章于 2023-07-27 11:25:43 发布

珍珠奶茶少冰半糖

最新推荐文章于 2023-07-27 11:25:43 发布

阅读量63

点赞数

分类专栏： LeetCode 文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/weixin_44399031/article/details/109679187

版权

LeetCode 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

该博客详细介绍了如何解决LeetCode第4题，即在两个已排序的数组中找到中位数。提出了一个时间复杂度为O(log(m+n))的解决方案，通过递归和二分查找策略，确定分割线并找到合适的中位数。示例展示了不同输入情况下算法的正确计算结果。

摘要由CSDN通过智能技术生成

寻找两个正序数组的中位数
LeetCode题库第4题

给定两个大小为 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
示例 3：
输入：nums1 = [0,0], nums2 = [0,0] 输出：0.00000
示例 4：
输入：nums1 = [], nums2 = [1] 输出：1.00000
示例 5：
输入：nums1 = [2], nums2 = [] 输出：2.00000

代码

public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
        //获取两个数组的长度
        int nums1Length = nums1.length;
        int nums2Length = nums2.length;

        //为了保证第一个数组为长度短的数组，进行判断递归
        if (nums1Length > nums2Length){
            findMedianSortedArrays(nums2,nums1);
        }

        /**取中位数的位置
         * 若nums1Length+nums2Length为奇数，则中位数为第（（nums1Length+nums2Length）+1）/2个元素
         * 若nums1Length+nums2Length为偶数，则中位数为（（nums1Length+nums2Length）+1）/2 及其前一个元素之和除以二（数组下标以0开头）
         * */
        //两个数组根据中位线分割之后，分割线左边应该有的元素总个数
        int leftTotalLength = ((nums1Length+nums2Length) + 1) / 2;

        int left = 0;
        int right = nums1Length;
        while(left < right){

            int i = left + (right-left+1)/2;
            int j = leftTotalLength - i;

            if(nums1[i-1] > nums2[j]){
                right = i - 1;
            }else{
                left = i;
            }
        }

        int i = left;
        int j = leftTotalLength - i;

        int nums1LeftMax = i == 0 ? Integer.MIN_VALUE : nums1[i-1];
        int nums1RightMin = i == nums1Length ? Integer.MAX_VALUE : nums1[i];
        int nums2LeftMax = j == 0 ? Integer.MIN_VALUE : nums2[j-1];
        int nums2RightMin = j == nums2Length ? Integer.MAX_VALUE : nums2[j];

        if((nums1Length+nums2Length)%2 == 1){
            return Math.max(nums1LeftMax,nums2LeftMax);
        }else{
            return (double)(Math.max(nums1LeftMax,nums2LeftMax)+Math.min(nums1RightMin,nums2RightMin))/2;
        }
    }