leetcode——腾讯企业真题--数组与字符串 2.寻找两个正序数组的中位数

给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

算法的时间复杂度应该为 O(log (m+n)) 。

      我自己做的未考虑时间复杂度,以解题的目的完成及完善了代码,如果要求时间复杂度的话,读者可以参考二分查找及其他的一些优秀的方法。

下面是我的思路:

先将两个数组合并成一个有序数组,再利用数学知识将中位数找出来。

合并两个数组用了双指针的方法,注意要将原数组中未存入的值存进新数组

找中位数:

如果新数组长度是奇数,则中位数正好在是中间,直接用新数组的长度对2取整即可;

若新数组长度是偶数,则中位数正好在是中间两个数的一半,所以我们需要找到中间的两个数。其中一个数是新数组的长度对2取整,另一个数是新数组的长度对2取整,再-1。

例:长度为5,中位数是5 / 2 = 2(下标是从0开始)

       长度为6,中间一个数是6 / 2 = 3,另一个数是6 / 2 - 1 = 2。再对这两个数进行操作即可。

代码如下:

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int[] mergeArray = new int[nums1.length + nums2.length];
        int index = 0;
        int i = 0, j = 0;
        if(mergeArray.length == 1){//合并后数长度为1,说明一个数组为空,一个数组只有一个元素
            if(nums1.length == 1){
                return nums1[0];
            }else if(nums2.length == 1){
                return nums2[0];
            }
        }

        
        //从小到大合并数组
        for( ; i < nums1.length && j < nums2.length;){
            if(nums1[i] < nums2[j]){
                mergeArray[index++] = nums1[i++];
            }else{
                mergeArray[index++] = nums2[j++];
            }
        }
        //存入未存进新数组的数
        if(i < nums1.length){
            for(; i < nums1.length;){
                mergeArray[index++] = nums1[i++];
            }
        }else if(j < nums2.length){
            for(; j < nums2.length;){
                mergeArray[index++] = nums2[j++];
            }
        }
        //分奇偶数获取中位数
        if((mergeArray.length & 1) == 0){//偶数
           double s1 = mergeArray[mergeArray.length / 2];
        	double s2 = mergeArray[(mergeArray.length / 2) - 1];
            return (s1 + s2) / 2;
        }else{
            return mergeArray[mergeArray.length / 2];
        }
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值