4.寻找两个正序数组的中位数
力扣题目链接:https://leetcode.cn/problems/median-of-two-sorted-arrays/
题意
给定两个大小分别为 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
提示
- nums1.length == m
- nums2.length == n
- 0 <= m <= 1000
- 0 <= n <= 1000
- 1 <= m + n <= 2000
- -106 <= nums1[i], nums2[i] <= 106
解题思路
由于两个数组都是有序的,因此只要从头开始比较直到找到中位数结束。
如果两数组长度为奇数则中位数只有一个,偶数则有两个求平均即可。
Java代码
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int n = nums1.length;
int m = nums2.length;
double sum = 0.0;
//i指向nums1数组,j指向nums2数组
for(int i=0,j=0;;){
//记录较小的值
int temp = 0;
//若指针越界则说明一个数组遍历完
if(i >= n){
temp=nums2[j++];
}else if(j >= m){
temp=nums1[i++];
}
else if(nums1[i]>nums2[j]){ //比较nums1[i]和nums2[j]遍历较小的值
temp=nums2[j++];
}else if(nums1[i]<=nums2[j]){
temp=nums1[i++];
}
//若为中位数则加在sum中,总长度为奇数则加一次,偶数加两次
if((i+j) >= (n+m+1)/2){
sum+=temp;
}
//遍历完中位数则退出循环
if((i+j) >= (n+m+2)/2){
break;
}
}
if(((n+m) & 1) ==0){
return sum/2;
}else{
return sum;
}
}
}