LeetCode-两个排序数组的中位数

本文介绍了一种求解两个有序数组中位数的方法,该方法的时间复杂度为O(log(m+n))。通过递归查找第k小元素,实现对不同长度数组中位数的有效计算。

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

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 

请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。

示例 1:

nums1 = [1, 3]
nums2 = [2]

中位数是 2.0

示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

中位数是 (2 + 3)/2 = 2.5


package subject4;

public class Subject4 {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Solution solution = new Solution();
		System.out.println(solution.findMedianSortedArrays(new int[] { 1, 2 }, new int[] { 3,4 }));
	}

}

class Solution {
	public double findMedianSortedArrays(int[] nums1, int[] nums2) {
		int length1 = nums1.length;
		int length2 = nums2.length;
        int length = length1+length2;
		
        if(length%2==1)
        {
        	return findKth(nums1, 0, length1, nums2, 0, length2, length / 2 + 1);
        }
        
		return (findKth(nums1, 0, length1, nums2, 0, length2, length/ 2)
				+ findKth(nums1, 0, length1, nums2, 0, length2, length / 2 + 1)) / 2;
	}

	private double findKth(int[] nums1, int start1, int length1, int[] nums2, int start2, int length2, int k) {

		//默认nums[]里面要处理的值比较少
		if(length1-start1>length2-start2)
		{
			return findKth(nums2, start2, length2, nums1, start1, length1, k);
		}
		
	
		//nums1[]里面的值都在前k中
		if(start1==length1)
		{
			return nums2[k-1];
		}
		
		if(k==1)
		{
			return Math.min(nums1[start1], nums2[start2]);
		}
		
	
		int nextStart1 = start1+Math.min(k/2, length1-start1);
		int nextStart2 = start2+(k-nextStart1+start1);
		
		
		if(nums1[nextStart1-1]<=nums2[nextStart2-1])
		{
			return findKth(nums1, nextStart1, length1, nums2, start2, length2, k-nextStart1+start1);
		}
		else 
		{
			return findKth(nums1, start1, length1, nums2, nextStart2, length2, k-nextStart2+start2);
		}
	

	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值