leetcode第四题:寻找两个有序数组的中位数

给定两个大小为 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

因为这个题目是两个有序数组,所以只要遍历两个数组的一半,找到中位数就可以,遍历的时候需要注意的是,要按照两个数组的从小到大来遍历,这样时间复杂度是(m+n)/2,也就是O(log(m+n)/2)

public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
    double res=0;
    int length=nums1.length+ nums2.length;
    int count=0,i=0,j=0;
//之所以定义两个temp,是因为奇数的中位数是中间那个数,偶数的中位数是中间两个数平均
    double temp1=0,temp2=0;
//以下一个循环主要是寻找对于偶数的第一个中位数,对于对于奇数是中位数的前一个数
        while(count<length/2){
//对四种情况分别进行判断,按照两个数组的从小到大来遍历
            if(i<nums1.length&&j<nums2.length&&nums1[i]<=nums2[j]){
               temp1=nums1[i];
                i++;
            }else if(i<nums1.length&&j<nums2.length&&nums1[i]>nums2[j]){
                temp1=nums2[j];
                j++;
            }else if(i<nums1.length&&j>=nums2.length){
                temp1=nums1[i];
                i++;
            }else{
                temp1=nums2[j];
                j++;
            }
            count++;
        }
//下面同理,这里还可以优化,如果上面是count<length/2+1的话,length是奇数是可以不用进行这一步的。
    if(i<nums1.length&&j<nums2.length&&nums1[i]<=nums2[j]){
        temp2=nums1[i];
    }else if(i<nums1.length&&j<nums2.length&&nums1[i]>nums2[j]){
        temp2=nums2[j];
    }else if(i<nums1.length&&j>=nums2.length){
        temp2=nums1[i];
    }else{
        temp2=nums2[j];
    }

  //  temp2=nums1[i]<=nums2[j]?nums1[i]:nums2[j];
    res=length%2==0?(temp1+temp2)/2:temp2;
    return res;

}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值