4.寻找两个正序数组的中位数
难度:困难
(1)题目复现
给定两个大小为 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
结果示例3:
输入:nums1 = [], nums2 = [1]
输出:1.00000
(2)解题思路
对于刷力扣没有任何经验的小白,我写的程序应该是挺笨的。
a.因为两个数组是有序的,所以首先就是将两个数组合并到一个数组中,用all数组存储nums1和nums2的合并结果;
b.题目的核心就是这个合并过程,首先用两个变量l1,l2分别作为两个数组的滑动变量,初始时都指向各数组的第一个位置,然后对nums1[l1]和nums2[l2]进行比较:
nums1[l1]<nums1[l2],nums1[l1]存储到all中,l1++,l2不变;
nums1[l1]=nums1[l2],nums1[l1]、nums2[l2]都存储到all中,l1++,l2++;
nums1[l1]>nums1[l2],nums2[l2]存储到all中,l1不变,l2++;
在l1和l2的移动过程中,有一方到达数组末尾,则停止,只需将未移动到末尾的另一个数组剩下的值存储到数组中即可;
(3)测试结果
(4)代码
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int len1 = nums1.length;//数组1长度
int len2 = nums2.length;//数组2长度
int[] all = new int[len1+len2];//合并后的数组为all
int k=0;
int l1=0,l2=0;//l1指向nums1,l2指向nums2
while(l1<len1&&l2<len2) {
if(nums1[l1]<nums2[l2]) {
all[k++] = nums1[l1];
l1++;
}else if(nums1[l1]==nums2[l2]){
all[k++] = nums1[l1];
all[k++] = nums2[l2];
l1++;
l2++;
}else{
all[k++] = nums2[l2];
l2++;
}
}
//上面的循环执行完后如果有数组没到头,则继续合并
if(l1!=len1) {
for(int i=l1;i<len1;i++) {
all[k++]=nums1[i];
}
}else if(l2!=len2){
for(int i=l2;i<len2;i++) {
all[k++]=nums2[i];
}
}
int length = len1+len2;
if((length)%2==0) return (double)(all[length/2-1]+all[length/2])/2;
else return (double)all[length/2];
}