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

letcode 两个排序数组的中位数

程序员文章站 2024-03-16 12:51:16
...

题目描述:
给定两个大小为 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[] num = merge(nums1,nums2);
        int N = num.length;
        double result=0;
        if(N%2==0) {
            result = (num[N/2]+num[N/2-1])/2.0;
        }else {
            result=num[N/2];
        }
        return result;
    }
    //用于归并数组,从小到大
    public  int[] merge(int[] nums1, int[] nums2) {
        int[] result = new int[nums1.length+nums2.length];
        if(nums1.length==0) return nums2;
        if(nums2.length==0) return nums1;
        int i=0;int j=0;
        int temp =0;
        while(i<nums1.length||j<nums2.length) {
            //归并小的那一个
            if(nums1[i]>nums2[j]) {
                result[temp++] = nums2[j++];
            }else {
                result[temp++] = nums1[i++];
            }
            //如果一个数组已经归并完归并另一个
            while(i>=nums1.length) {
                if(j>=nums2.length) break;
                result[temp++] = nums2[j++];
            }
            while(j>=nums2.length) {
                if(i>=nums1.length) break;
                result[temp++] = nums1[i++];
            }
        }
        return result;
    }
}
相关标签: letcode