题目:给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log (m+n)) 。
输入:nums1 = [1,3], nums2 = [2] 输出:2.00000 解释:合并数组 = [1,2,3] ,中位数 2
困难难度,难度在于题目设置了时间复杂度为 O(log (m+n)),所以加大了难度,一般的暴力解法肯定不行。
解题思路:先合并数组再进行中位数。
难处就在合并数组,题目可知两个数组都是正序的,难度减小了很多,那么同样需要两个变量i和j,如果第二个数组nums2[j]大于第一个数组nums1[i],那么将小的(即数组一nums1[i])放入数组三中,同时i++,否则将nums2[j]放入数组三j++,循环判断nums2[j]和此时下一个的nums1的大小。最后将已排好序的剩下的元素全部添加到数组三中
//解题思路:1.合并两个数组 2.寻找中位数
int i=0,j=0,k=0;
int m = nums1.length;int n = nums2.length;
int[] nums3 = new int[m+n];
while(i<m && j<n){
if(nums2[j]>nums1[i]){
nums3[k] = nums1[i];
i++;
}else{
nums3[k] = nums2[j];
j++;
}
k++;
}
while(i < m){
nums3[k]=nums1[i];
i++;
k++;
}
while(j < n){
nums3[k]=nums2[j];
j++;
k++;
}
if(nums3.length%2==0){
return (double)(nums3[nums3.length/2]+nums3[nums3.length/2-1])/2;
}else return nums3[(nums3.length-1)/2];