目录
题目:
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
示例 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 = [0,0], nums2 = [0,0]
输出:0.00000
示例 4:输入:nums1 = [], nums2 = [1]
输出:1.00000
示例 5:输入:nums1 = [2], nums2 = []
输出:2.00000来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays
思路:
拿到这道题,暴力解法就是合并数组,排序后取中位数,显然这样时间复杂度太高。
我们这里的思路用的是
1.首先根据输入分为2中情况
(1)输入的数组有空的(这种要考虑,不然会空指针异常)
(2)都不为空。
第一种,情况只需要在不为空的数组中找中位数即可,我们针对第二种情况详细讲解
2.我们这里可以理解为从右边往左边拿数据(不是真拿,只是用m,n这两个索引作为分界线)用temp1,保存上一拿到的,temp2保存上上个拿到的,当拿到的数据占一半多一个时停止,此时(如果总数为基数,temp1保存的就是结果,因为9/2=4,我们需要第五个)(如果总数为偶数,temp1和temp2的平均数就是结果,因为10/2=5,我们需要的就是第5个和下一个,即第六个)。
代码:
package LeetCode;
public class ZhengXuZhongWeiShu {
public static void main(String[] args) {
Solution solution = new Solution();
int[] a = {};
int[] b = {2,3};
double d=solution.findMedianSortedArrays(a, b);
System.out.println(d);
}
}
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int length = nums1.length + nums2.length;
//先判断num1和num2有没有空的,有就直接返回另一个的中位数。
if (nums1.length == 0) {
return length % 2 != 0 ? nums2[length / 2] : ((double) (nums2[length / 2] + nums2[length / 2 - 1]) / 2);
}
if (nums2.length == 0) {
return length % 2 != 0 ? nums1[length / 2] : ((double) (nums1[length / 2] + nums1[length / 2 - 1]) / 2);
}
int temp1=0;//记录最后一个元素
int temp2=0;//记录倒数第二个元素
//当两者都不为空时
int m = 0, n = 0;//m为nums1的索引,n为nums2的索引
while (m + n <= length / 2 ) {
if(m==nums1.length){
temp2=temp1;
temp1=nums2[n];
n++;
}else if(n==nums2.length){
temp2=temp1;
temp1=nums1[m];
m++;
}else if (nums1[m] < nums2[n]) {
temp2=temp1;
temp1=nums1[m];
m++;
}else{
temp2=temp1;
temp1=nums2[n];
n++;
}
}
if(length%2==0){
return (double) (temp1+temp2)/2;
}else{
return (double) temp1;
}
}
}
结果: