【LeetCode-4】Median of Two Sorted Arrays

最新推荐文章于 2023-06-12 16:30:30 发布

转载最新推荐文章于 2023-06-12 16:30:30 发布 · 216 阅读

文章标签：

#LeetCode-4 #Median of Two Sorted #K小值

leetcode 专栏收录该内容

80 篇文章

订阅专栏

本文介绍了一种使用查找第K小值的方法来找到两个已排序数组的中位数。该方法首先判断总元素数量的奇偶性，然后通过递归调用findKth函数实现。此算法的关键在于正确实现findKth函数，它能够有效地处理不同长度数组间的比较，并返回所需位置的数值。

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

这道题好难，一点没想到用求第K个最小值的办法来解决，共勉吧！

public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int total = nums1.length + nums2.length;
        if (total % 2 == 1)
            return findKth(nums1, 0, nums2, 0, total / 2 + 1);
        else
            return (findKth(nums1, 0, nums2, 0, total / 2) + findKth(nums1, 0, nums2, 0, total / 2 + 1)) / 2;
    }
 
	//k代表第k个最小的数,关键就在这个函数
    private double findKth(int[] a, int abegin, int[] b, int bbegin, int k) {
        if (a.length > b.length)
            return findKth(b, bbegin, a, abegin, k);
 
        if (a.length == 0)
            return b[k - 1];
 
        if (k == 1)
            return Math.min(a[0], b[0]);
 
        int pa = Math.min(k / 2, a.length), pb = k - pa;
        
        if (a[pa - 1] < b[pb - 1])
            return findKth(Arrays.copyOfRange(a, pa, a.length), abegin + pa, b, bbegin, k - pa);
        else if (a[pa - 1] > b[pb - 1])
            return findKth(a, abegin, Arrays.copyOfRange(b, pb, b.length), bbegin + pb, k - pb);
        else
            return a[pa - 1];
    }