题目:
给定两个大小分别为
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
思路一:
暴力解法,先将这两个数组进行合并,然后对合并了的数组进行排序;
最后通过下面这个式子
((m+n)%2==0)?(num[(m+n)/2]+num[(m+n)/2-1])/2.0:(double)num[(m+n)/2];
进行计算中位数。详细代码如下:
import java.util.*;
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int m = nums1.length;int n = nums2.length;
int[] num = new int[m + n];
int i = 0;int j = 0;
for(i = 0;i<m;i++) {
num[i] = nums1[i];
}
for(j = 0;j < n;j++) {
num[i++] = nums2[j];
}
Arrays.sort(num);
return ((m+n)%2==0)?(num[(m+n)/2]+num[(m+n)/2-1])/2.0:(double)num[(m+n)/2];
}
}
思路二:
可以使用一个类似于双指针的方法进行计算;首先,定义一个 i 是用来遍历数组nums1的,定义一个 j 是用来遍历数组nums2的。题目要求计算中位数,可以分为两种情况,一种为两个数组总和为奇数个的时候,中位数是 下标(m + n)/ 2对应的数;第二种情况是两个数组总和为偶数个的时候,中位数是 下标(m + n)/ 2 和 下标(m + n)/ 2-1对应的数字之和除以 2;所以需要循环(m + n)/ 2 次去寻找中位数,同时记录下第(m + n)/ 2个数字和第(m + n)/ 2 -1 个数字;用num1 去记录每一次被跳过的那个值,用num2 记录第(m + n)/ 2 -1 个数字,用num3记录第(m + n)/ 2个数字。会出现一个数组被遍历完然而还没有找到中位数的情况;详细代码如下:(切记if条件语句中的判断条件不能错位,同时else语句不能省略,会出现下标越界的情况)
import java.util.*;
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int count = 0;
int m = nums1.length;int n = nums2.length;
int i = 0;int j = 0;
int num1 = 0;int num2 = 0;int num3 = 0;
for(count = 0;count <= (m + n)/2;count++) {
if(i < nums1.length && j < nums2.length && nums1[i] < nums2[j]){num1 = nums1[i++];}
else if(j < nums2.length && i < nums1.length && nums1[i] >= nums2[j]) num1 = nums2[j++];
else if(i == nums1.length) {num1 = nums2[j++];}
else if(j == nums2.length) {num1 = nums1[i++];}
if(count == (m + n) / 2 -1) num2 = num1;
if(count == (m + n) / 2) num3 = num1;
}
if((m + n) % 2 != 0) return num3;
else return (num3 + num2 ) / 2.0;
}
}