leetcode4 寻找两个有序数组的中位数
描述
给定两个大小为 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
难度
中,归并排序
解题思路
利用归并排序的思想,数组的总个数为偶数时,寻找第(l1+l2)/2和(l1+l2)/2+1个数 。为奇数时,寻找第 (l1+l2)/2+1个数。
代码
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int l1 = nums1.length;
int l2 = nums2.length;
int k=0,a=0,b=0,count=0;
int i=0,j=0;
if((l1+l2)%2==0){
while(i<l1 && j<l2){
if(nums1[i]<=nums2[j]){
count++;
i++;
if(count==(l1+l2)/2) a = nums1[i-1];
if(count==(l1+l2)/2+1) b = nums1[i-1];
}else{
count++;
j++;
if(count==(l1+l2)/2) a = nums2[j-1];
if(count==(l1+l2)/2+1) b = nums2[j-1];
}
if(a!=0 && b!=0) return((a+b)/2.0);
}
while(i<l1){
count++;
i++;
if(count==(l1+l2)/2) a = nums1[i-1];
if(count==(l1+l2)/2+1) b = nums1[i-1];
}
while(j<l2){
count++;
j++;
if(count==(l1+l2)/2) a = nums2[j-1];
if(count==(l1+l2)/2+1) b = nums2[j-1];
}
return((a+b)/2.0);
}
else{
while(i<l1 && j<l2){
if(nums1[i]<=nums2[j]){
count++;
i++;
if(count==(l1+l2)/2+1) k = nums1[i-1];
}else{
count++;
j++;
if(count==(l1+l2)/2+1) k = nums2[j-1];
}
if(k!=0) return(k);
}
while(i<l1){
count++;
i++;
if(count==(l1+l2)/2+1) k = nums1[i-1];
}
while(j<l2){
count++;
j++;
if(count==(l1+l2)/2+1) k = nums2[j-1];
}
return k;
}
}
}