leetcode：Median of Two Sorted Arrays 【Java】

最新推荐文章于 2023-06-12 16:30:30 发布

tterminator

最新推荐文章于 2023-06-12 16:30:30 发布

阅读量353

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/tterminator/article/details/50830773

leetcode 专栏收录该内容

79 篇文章

订阅专栏

本文介绍了一种算法，用于找到两个已排序数组的中位数。通过合并这两个数组并利用数组已排序的特点来实现O(log(m+n))的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、问题描述

There are two sorted arrays nums1 and nums2 of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

二、问题分析

合并两个有序数组（参考算法leetcode：Merge Sorted Array 【Java】）后，寻找中位数。

三、算法代码

public class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int [] mergedArray = merge(nums1, nums1.length, nums2, nums2.length);
        int len = mergedArray.length;

        double middle = 0.0;
        if(len % 2 == 0){
            middle = ((double)mergedArray[(len - 1) / 2] + (double)mergedArray[(len - 1) / 2 + 1]) / 2;
        }else{
            middle = (double)mergedArray[(len - 1) / 2];
        }
        return middle;
    }
    
    //合并两个有序数组
    public int[] merge(int[] nums1, int m, int[] nums2, int n) {
        int a_index = m - 1;
        int b_index = n - 1;
        int a_cur = m + n - 1;
        int [] result = new int[m + n];
        while(a_index >= 0 && b_index >= 0){
        	if(nums1[a_index] >= nums2[b_index]){
        		result[a_cur] = nums1[a_index];
        		a_cur--;
        		a_index--;
        	}else{
        		result[a_cur] = nums2[b_index];
        		a_cur--;
        		b_index--;
        	}
        }
        while(b_index >= 0){
        	result[a_cur] = nums2[b_index];
    		a_cur--;
    		b_index--;
        }
        while(a_index >= 0){
        	result[a_cur] = nums1[a_index];
    		a_cur--;
    		a_index--;
        }
        return result;
    }
}