题目描述:
给定两个大小为 m 和 n 的正序(从小到大)数组 nums1
和 nums2
。
请你找出这两个正序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1
和 nums2
不会同时为空。
示例 1:
nums1 = [1, 3] nums2 = [2] 则中位数是 2.0
因为两个数组都是正序数组,所以中位数可以直接根据顺序找到中间的一个或两个数字(根据奇偶区别)即可。
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
double res = 0;
int l1 = nums1.length,l2 = nums2.length;
//判断奇偶
int ex = (l1+l2+1)%2;
//记录上一个数,中位数需要取两个时使用
double temp = 0;
int n1=0,n2=0;
for(int i=0;i<=(l1+l2)/2;i++){
//每次去较小的那个数即可
if(n1<l1&&n2<l2){
if(nums1[n1]<nums2[n2]){
temp = res;
res = nums1[n1];
n1++;
}else{
temp = res;
res = nums2[n2];
n2++;
}
}else{
if(n1<l1){
temp = res;
res = nums1[n1];
n1++;
}else{
temp = res;
res = nums2[n2];
n2++;
}
}
}
if(ex>0){
res = (temp + res)/2;
}
return res;
}
}
时间复杂度(m+n)/2 空间复杂度(1)