LeetCode-004-寻找两个正序数组的中位数-优快云博客

这篇博客介绍了如何利用二分查找算法在O(log(m+n))的时间复杂度内找到两个正序数组的中位数。通过getKthElement方法获取第k小的元素，然后根据数组长度的奇偶性计算中位数。

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

LeetCode-004-寻找两个正序数组的中位数

在这里插入图片描述

思路

题目要求时间复杂度为 O(log (m+n))，那么首先想到的就是二分查找

代码

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int len1 = nums1.length, len2 = nums2.length;
        int len = len1 + len2;
        if (len % 2 == 1) {
            double mid = getKthElement(nums1, nums2, len/2 + 1);
            return mid;
        } else {
            double mid = (getKthElement(nums1, nums2, len/2) + getKthElement(nums1, nums2, len/2 + 1)) / 2.0;
            return mid;
        }
    }

    public double getKthElement(int[] nums1, int[] nums2, int k) {
        int len1 = nums1.length, len2 = nums2.length;
        int idx1 = 0, idx2 = 0;

        while(true){
            if(idx1==len1) return nums2[idx2+k-1];
            if(idx2==len2) return nums1[idx1+k-1];
            if(k==1) return Math.min(nums1[idx1],nums2[idx2]);

            int newIdx1 = Math.min(idx1 + k/2, len1) - 1;
            int newIdx2 = Math.min(idx2 + k/2, len2) - 1;
            int tmp1 = nums1[newIdx1], tmp2 = nums2[newIdx2];
            if (tmp1 <= tmp2) {
                k -= (newIdx1 - idx1 + 1);
                idx1 = newIdx1 + 1;
            } else {
                k -= (newIdx2 - idx2 + 1);
                idx2 = newIdx2 + 1;
            }
        }
    }
}