给定两个大小分别为 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
这道题难度是困难,感觉名不副实啊,解题的话
1、较为省心的办法是将两数组合并,然后取中位数,时间复杂度O(log (m+n)) 代码简洁美观,可读性好。
2、在遍历到中位数时直接取值时间复杂度<O(log (m+n)),由于加了过多逻辑会导致代码量过高,不易阅读
1.
int[] num = new int[nums1.length+nums2.length];
int n1 = 1,n2=1;
for(int i = num.length-1;i>=0;i--){
if(n1<=nums1.length&&n2<=nums2.length){
if(nums1[nums1.length-n1]>nums2[nums2.length-n2]){
num[i] = nums1[nums1.length-n1];n1++;
}else {
num[i] = nums2[nums2.length-n2];n2++;
}
}else{
if(n1<=nums1.length){
num[i] = nums1[nums1.length-n1];n1++;
continue;
}else {
num[i] = nums2[nums2.length-n2];n2++;
continue;
}
}
}
if(num.length%2==0){
int length = num.length/2;
return (num[length-1]+num[length])/2.0000;
}else{
return num[num.length/2];
}
2、
int[] num = new int[nums1.length+nums2.length];
int n1 = 1,n2=1;
if(num.length==1){
if(nums2.length>0){
return nums2[0];
}else {
return nums1[0];
}
}
for(int i = num.length-1;i>=0;i--){
if(n1<=nums1.length&&n2<=nums2.length){
if(nums1[nums1.length-n1]>nums2[nums2.length-n2]){
num[i] = nums1[nums1.length-n1];n1++;
}else {
num[i] = nums2[nums2.length-n2];n2++;
}
}else{
if(n1<=nums1.length){
num[i] = nums1[nums1.length-n1];n1++;
}else {
num[i] = nums2[nums2.length-n2];n2++;
}
}
if(i<num.length/2){
if(num.length%2==0){
int length = num.length/2;
return (num[length-1]+num[length])/2.0000;
}else{
return num[num.length/2];
}
}
}
return 0;