leetcode4. 寻找两个正序数组的中位数

最新推荐文章于 2026-01-03 14:34:08 发布

原创最新推荐文章于 2026-01-03 14:34:08 发布 · 1.5k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#指针 #leetcode #java

Java面试专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种在两个已排序数组中查找中位数的方法，通过双指针法实现，确保了O(log(m+n))的时间复杂度。文章详细解析了算法步骤，包括移动次数的计算、指针更新及最终中位数的确定。

题目

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

解题思路

双指针法，每次比较两个数组指定位置的大小并移动指针位置

代码

public  double findMedianSortedArrays(int[] nums1, int[] nums2) {
    int nums1Length = nums1.length;
    int nums2Length = nums2.length;

	//移动的次数，偶数减1
    int centre = (nums1Length + nums2Length) % 2 == 0 ? (nums1Length + nums2Length) / 2 - 1 : (nums1Length + nums2Length) / 2;

    int nums1Left = 0;
    int nums2Left = 0;
    while(centre > 0) {
        int value1 = nums1Left == nums1Length ? Integer.MAX_VALUE : nums1[nums1Left];
        int value2 = nums2Left == nums2Length ? Integer.MAX_VALUE : nums2[nums2Left];
        if (value1 < value2) {
            nums1Left++;
        } else {
            nums2Left++;
        }
        centre--;
    }

	
    int first = (nums1Left == nums1Length || (nums2Left != nums2Length && nums1[nums1Left] > nums2[nums2Left]))
            ? nums2[nums2Left] : nums1[nums1Left];
    //两个数组长度和为偶数时
    if ((nums1Length + nums2Length) % 2 == 0) {
    	//如果nums1Left没有走到最后，并且first值等于nums1[nums1Left]，表示first是在nums1数组
        if (nums1Left != nums1Length && first == nums1[nums1Left]) {
            nums1Left++;
        } else {
            nums2Left++;
        }
        int second = (nums1Left == nums1Length || (nums2Left != nums2Length && nums1[nums1Left] > nums2[nums2Left]))
                ? nums2[nums2Left] : nums1[nums1Left];
        return (first + second) / 2.0;
    }
    return (double)first;
}