leetcode刷题4(寻找两个有序数组的中位数)

题目描述:

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

通过答案:

class Solution {               //直接暴力求解
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m=nums1.length;
        int n=nums2.length;
        int[] nums;            //声明一个新数组用来存放两个数组合并后的情景
        nums=new int[m+n];
        int i=0;
        int j=0;
        int count=0;   
        if(m==0){             //若nums1数组为空
            if(n%2==0){       //若nums2数组长度为偶数
                return (nums2[n/2-1]+nums2[n/2])/2.0;     //中位数为最中间两个数的平均
            }else{
                return nums2[n/2];
            }
        }
        if(n==0){              //若nums2数组为空
            if(m%2==0){
                return (nums1[m/2-1]+nums1[m/2])/2.0;
            }else{
                return nums1[m/2];
            }
        }
        while(count!=m+n){        //若新数组长度不为两数组合并后的长度
            if(i==m){             //nums1已经遍历完
                while(j!=n){
                    nums[count++]=nums2[j++];      //直接将nums2剩下的数放到新数组的后面
                }
                break;
            }
            if(j==n){              //若nums2已经遍历完
                while(i!=m){
                    nums[count++]=nums1[i++];
                }
                break;
            }
            if(nums1[i]>nums2[j]){     //若nums1中的数大于nums2中的数
                nums[count++]=nums2[j++];    //将小的数加入到新的数组
            }else{
                nums[count++]=nums1[i++];
            }
        }
        if(count%2==0){            //若合并后的新数组的长度为偶数
            return (nums[count/2-1]+nums[count/2])/2.0;
        }else{
            return nums[count/2];
        }
        
    }
}
发布了88 篇原创文章 · 获赞 10 · 访问量 2556

猜你喜欢

转载自blog.csdn.net/ZhangShaoYan111/article/details/104562878