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

最新推荐文章于 2025-02-05 21:00:00 发布

原创最新推荐文章于 2025-02-05 21:00:00 发布 · 97 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种算法，用于查找两个已排序数组的中位数。该算法首先将两个数组合并，并对合并后的数组进行排序，然后根据数组长度的奇偶性计算并返回中位数。

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

给定两个大小为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。

进阶：你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗？

示例 1：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

示例 3：

输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000

示例 4：

输入：nums1 = [], nums2 = [1]
输出：1.00000

示例 5：

输入：nums1 = [2], nums2 = []
输出：2.00000

提示：

nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106

import java.util.Arrays;

public class Solution4 {
	public double findMedianSortedArrays(int[] nums1, int[] nums2) {

		double out = 1;

		int[] a = new int[nums1.length + nums2.length];
		System.arraycopy(nums1, 0, a, 0, nums1.length);
		System.arraycopy(nums2, 0, a, nums1.length, nums2.length);
		Arrays.sort(a);
		// System.out.println(Arrays.toString(a));
		if (a.length % 2 == 0) {

			out = ((double) a[a.length / 2 - 1] + (double) a[a.length / 2]) / 2;
		} else {
			out = (double) a[a.length / 2];
		}

		return out;
	}

	public static void main(String[] args) {

		Solution4 s = new Solution4();

		int[] nums1 = { 1, 3 };
		int[] nums2 = { 2 };

		for (int n : nums1) {
			System.out.println(n);
		}

		System.out.println(s.findMedianSortedArrays(nums1, nums2));
	}
}