欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-05-20 12:27:49
...

题目描述:

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