LeetCode 4. 寻找两个正序数组的中位数
题目
4. 寻找两个正序数组的中位数
给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。
请你找出这两个正序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。
示例 1:
nums1 = [1, 3]
nums2 = [2]
则中位数是 2.0
示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
则中位数是 (2 + 3)/2 = 2.5
思路,双指针 O(n)复杂度
代码
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int len =nums1.length;
len+=nums2.length;
int [] list=new int[len];
int i ,j,k;
i=j=k=0;
while( i<nums1.length&&j<nums2.length){
list[k++]=(nums1[i]>nums2[j] ? nums2[j++]:nums1[i++]);
}
while(i<nums1.length)list[k++]=nums1[i++];
while(j<nums2.length)list[k++]=nums2[j++];
double v=0;
if((len&1)==1){
v= 2*list[len/2];
}else{
v=list[len/2]+list[len/2-1];
}
return v/2;
}
}