LeetCode 004 寻找两个有序数组的中位数

最新推荐文章于 2021-10-12 00:00:00 发布

原创最新推荐文章于 2021-10-12 00:00:00 发布 · 194 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

LeetCode 专栏收录该内容

79 篇文章

订阅专栏

本文介绍了一种求解两个有序数组中位数的方法，通过合并两个数组并利用时间复杂度为O(log(m+n))的算法高效计算中位数。示例展示了不同情况下的计算过程。

004 寻找两个有序数组的中位数
给定两个大小为 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

代码：
class Solution {
public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
int m=nums1.length;
int n=nums2.length;
int i=0,j=0,k=0;
int[] nums3=new int[(m+n)/2+1];
while(i<m&&j<n&&i+j<nums3.length){
if(nums1[i]<nums2[j]){
nums3[k++]=nums1[i++];
}else{
nums3[k++]=nums2[j++];
}
}
while(i<m&&i+j<nums3.length){
nums3[k++]=nums1[i++];
}
while(j<n&&i+j<nums3.length){
nums3[k++]=nums2[j++];
}
if((m+n)%2==0){
return ((double)(nums3[(m+n)/2-1]+nums3[(m+n)/2]))/2;
}else{
return (double)(nums3[(m+n)/2]);
}
}
}