文章目录
- 题目描述
- 方法一,重组排序
- 方法二,调用系统函数
题目描述
给定两个大小分别为 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
提示:
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法一,重组排序
思路:将两个数组合并到另一个大数组中.其实可以再优化,在排序时采用更高效算法即可.
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int len1=nums1.length;
int len2=nums2.length;
int[] arr=new int[len1+len2];
int k=0;
int i,j;
for( i=0,j=0;i<len1&&j<len2;){
if(nums1[i]<=nums2[j]){
arr[k]=nums1[i];
i++;
}else{
arr[k]=nums2[j];
j++;
}
k++;
}
while(i<len1){
arr[k++]=nums1[i++];
}
while(j<len2){
arr[k++]=nums2[j++];
}
if(arr.length%2==1){
return arr[arr.length/2];
}else{
//这里值得注意,容易出错
return (arr[(arr.length-1)/2]+arr[arr.length/2])/2.0;
}
}
}
方法二,调用系统函数
其实思路和方法一相同,只是系统已有数组拷贝函数和排序函数,故尝试调用,但似乎很耗时.
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int len1=nums1.length;
int len2=nums2.length;
int[] arr=new int[len1+len2];
System.arraycopy(nums1,0,arr,0,len1);
System.arraycopy(nums2,0,arr,len1,len2);
Arrays.sort(arr);
if(arr.length%2==1){
return arr[arr.length/2];
}else{
return (arr[(arr.length-1)/2]+arr[arr.length/2])/2.0;
}
}
}