LeetCode4寻找两个正序数组的中位数

本文探讨了如何在给定两个已排序数组中找到中位数,利用二分查找法实现时间复杂度为O(log(m+n))的高效算法。两种暴力破解解法对比,以及一种改进的搜索策略,适用于中位数求解的实际场景。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

描述:给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。算法的时间复杂度应该为 O(log (m+n)) 。

部分来自:LeetCode

解法1:暴力破解

思路:找到第len/2+1,或者len/2和len/2+1的数

 public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int length=nums1.length+nums2.length;
        double middle=0;
        int index=0;//总的数组的下标
        int i=0,j=0;
        int temp;//暂时存放当前的值
        if (length%2==0)//偶数的情况,即要求array[length/2-1]和array[length/2]的平均数
        {
            boolean flag=false;
            int num1=0,num2 = 0;
            for (;i<nums1.length&&j<nums2.length;)
            {
                if (nums1[i]<=nums2[j]){
                    temp=nums1[i];
                    i++;
                }
                else {
                    temp=nums2[j];
                    j++;
                }
                if (index==length/2-1)num1=temp;
                if (index==length/2) {
                    num2 = temp;
                    flag=true;//代表已经拿到
                    break;
                }
                index++;
            }

            if (flag==false) {
                if (index - 1 == length / 2 - 1)//此时num1已经写入,但num2并没有
                {
                    if (i == nums1.length)
                        num2 = nums2[j];
                    else num2 = nums1[i];
                } else if (index - 1 < length / 2 - 1) {//对应2个都没有被写入的情况
                    if (i == nums1.length) {
                        num1 = nums2[j + length / 2 - 1 - index];
                        num2 = nums2[j + length / 2 - index];
                    } else {
                        num1 = nums1[i + length / 2 - 1 - index];
                        num2 = nums1[i + length / 2 - index];
                    }
                }
            }
            middle=((double) num1+(double)num2)/2;
        }
        else {//奇数的情况,即要的是array[length/2]
            boolean flag=false;//若为true,则已经获得middle
            for (;i<nums1.length&&j<nums2.length;)
            {
                if (nums1[i]<=nums2[j]){
                    temp=nums1[i];
                    i++;
                }
                else {
                    temp=nums2[j];
                    j++;
                }
                if (index==length/2) {
                    middle=temp;
                    flag=true;
                    break;
                }
                index++;
            }
            if (flag==false)
            {
                if (i==nums1.length)
                middle=nums2[length/2-index+j];
                else middle=nums1[length/2-index+i];
            }
        }
       return middle;
}

这里有个小问题,处理特殊情况时,要考虑情况是否重合。最好的办法是如果已经拿到所需的值,用一个flag屏蔽后面的步骤。

解法二:暴力破解

思路:只需要找到中位数,同解法一

public double findMedianSortedArrays(int[] A, int[] B) {
    int m = A.length;
    int n = B.length;
    int len = m + n;
    int left = -1, right = -1;
    int aStart = 0, bStart = 0;
    for (int i = 0; i <= len / 2; i++) {
        left = right;
        if (aStart < m && (bStart >= n || A[aStart] < B[bStart])) {//重点
            right = A[aStart++];
        } else {
            right = B[bStart++];
        }
    }
    if ((len & 1) == 0)
        return (left + right) / 2.0;
    else
        return right;
}

重点看此解法中找中位数的方式,很简单,比解法一更好。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值