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

原创于 2019-08-05 23:39:41 发布 · 88 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode

学习同时被 2 个专栏收录

20 篇文章

订阅专栏

算法

13 篇文章

订阅专栏

本文探讨了在两个已排序的数组中查找中位数的问题，提出了一种时间复杂度为O(n)的方法，强调了二分法和分治思想的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空

我的答案

public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int l = nums1.length + nums2.length;
        int[] arr = new int[l/2+1];
        int i=0,j=0,k=0;
        double result =0;
        while (k<arr.length){
            if (i == nums1.length){
                arr[k] = nums2[j];
                j++;
            }else if (j== nums2.length){
                arr[k] = nums1[i];
                i++;
            }else if (nums1[i] <= nums2[j]){
                arr[k] = nums1[i];
                i++;
            }else {
                arr[k] = nums2[j];
                j++;
            }
            k++;
        }
        if (l%2 == 1){
            result = arr[k-1];
        }else {
            result = (arr[k-1] + arr[k-2])/2.0;
        }
        return result;
    }

总结
时间复杂度为O(n)
要学会使用二分法，分治思想。
最关键的点在于找到终止条件