力扣4.寻找两个正序数组的中位数(通过归并解决)

楚钧艾克

已于 2024-05-27 15:37:58 修改

阅读量406

点赞数 10

分类专栏：后端 # Java 文章标签： leetcode 算法职场和发展

于 2024-01-15 11:40:39 首次发布

本文链接：https://blog.csdn.net/c0210g/article/details/135597620

版权

后端同时被 2 个专栏收录

9 篇文章 2 订阅

订阅专栏

Java

7 篇文章 0 订阅

订阅专栏

本文介绍了一个编程问题，如何找到两个已排序数组的中位数，通过合并两个数组、排序和处理奇偶数组长度来实现。作者提供了Solution类中的findMedianSortedArrays方法，利用递归和双指针技巧完成数组操作。

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

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

代码:

class Solution 
{
    public double findMedianSortedArrays(int[] nums1 , int[] nums2)
    {
        //合并数组
        int[] nums = new int[nums1.length+nums2.length];
        int i = 0;
        for(; i < nums1.length ; i++)
        {
            nums[i] = nums1[i];
        }
        for(int index = 0; index < nums2.length ; index++)
        {
            nums[i++] = nums2[index];
        }
        int left = 0;
        int right = nums.length -1;
        //对数组进行排序
        process(nums,left,right);
        //求数组的中位数
        //如果数组长度为偶数
        double mid = 0;
        if(nums.length % 2 == 0)
        {
            mid =( nums[nums.length / 2] + nums[(nums.length / 2) - 1] ) / 2.0;
        }
        else
        {
            mid = nums[nums.length / 2];
        }
        return mid;
    }
    public static void process(int[] nums , int left , int right)
    {
        if(left == right)
        {
            return;
        }
        int mid = (left + right) >> 1;
        //对数组左侧进行排序
        process(nums,left,mid);
        //对数组右侧进行排序
        process(nums,mid+1,right);
        //对数组整体进行排序
        merge(nums,left,mid,right);
    }
    public static void merge(int[] nums , int left , int mid , int right)
    {
        int i = 0;
        int p1 = left;
        int p2 = mid + 1;
        int[] help = new int[right-left+1];
        while(p1 <= mid && p2 <= right)
        {
            help[i++] = nums[p1] < nums[p2] ? nums[p1++] : nums[p2++];
        }
        //检测左侧数组
        while(p1 <= mid)
        {
            help[i++] = nums[p1++];
        }
        //检测右侧数组
        while(p2 <= right)
        {
            help[i++] = nums[p2++];
        }
        //原地修改数组
        for(i = 0; i < help.length ; i++)
        {
            nums[i+left] = help[i];
        }
    }
}