寻找两个有序数组的中位数

题目:
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。

请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

方法一:
1.中位数:偶数个中间两个和除以2,奇数个中位数为中间那个
2.先判断是否其中有一个数组为空,如果有直接返回中位数即可
3.将两个数组中的值重新排序到新的数组,返回新数组的中位数

class FindMedianSortedArrays {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int len1=nums1.length;
        int len2=nums2.length;
        if(len1==0){
            if(len2%2==0){
                return (nums2[len2/2-1]+nums2[len2/2])/2.0;
            }else{
                return nums2[len2/2];
            }
        }
        if(len2==0){
            if(len1%2==0){
                return (nums1[len1/2-1]+nums1[len1/2])/2.0;
            }else{
                return nums1[len1/2];
            }
        }
        int[] arr=new int[len1+len2];
        int count=0;
        int m=0;
        int n=0;
        while(count!=(len1+len2)){
            if(m==len1){
                while(n!=len2){
                    arr[count++]=nums2[n++];
                }
                break;
            }
            if(n==len2){
                while(m!=len1){
                    arr[count++]=nums1[m++];
                }
                break;
            }
            if(nums1[m]>nums2[n]){
                arr[count++]=nums2[n++];
            }else{
                arr[count++]=nums1[m++];
            }
        }
        int len=len1+len2;
        if((len)%2==0){
            return (arr[len/2-1]+arr[len/2])/2.0;
        }else{
            return arr[len/2];
        }
    }
}

方法二:
不需要将两个数组真的合并,我们只需要找到中位数在哪里就可以了
1.用 len 表示合并后数组的长度,如果是奇数,我们需要知道第 (len+1)/2 个数就可以了,如果遍历的话需要遍历 int(len/2 ) + 1 次。如果是偶数,我们需要知道第 len/2和 len/2+1 个数,也是需要遍历 len/2+1 次。所以遍历的话,奇数和偶数都是 len/2+1 次。
2.用两个变量 left 和 right,right 保存当前循环的结果,在每次循环前将 right 的值赋给 left。这样在最后一次循环的时候,left 将得到 right 的值,也就是上一次循环的结果,接下来 right 更新为最后一次的结果。
3.用aflag标志nums1的下标,用bflag标志nums2的下标

class FindMedianSortedArrays1 {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int len1=nums1.length;
        int len2=nums2.length;
        int len=len1+len2;
        int aflag=0;
        int bflag=0;
        int left=0;
        int right=0;
        for(int i=0;i<=len/2;i++){
            left=right;
            if(aflag<len1&&(bflag>=len2||nums2[bflag]>nums1[aflag])){
                right=nums1[aflag++];
            }else{
                right=nums2[bflag++];
            }
        }
        if((len&1)==0){
            return (left+right)/2.0;
        }else{
            return right;
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值