寻找两个有序数组的中位数(更新一种解决方案)

11 篇文章 0 订阅
3 篇文章 0 订阅

题目见https://blog.csdn.net/qq_35684046/article/details/104330155

法一:
我们用两个指针i,j,分别指向n1与n2
遍历(len1+len2+1)/2次后跳出循环。
pre = now;
(1)如果i<n1.lenth&&(j>=n2.lenth||n1[i]>n2[i]),令now = n1[i++]。
(2)其他情况下,now = n2[j++];

如果两数组合并后为奇数返回now,否则返回(pre+now)/2。
时间复杂度为O((m+n)/2),并没有达到题目的要求,所以我们尝试法二。

法二:
我们猜想,如果要达到题目所要求的O(log(m+n)),可能需要用到二分法。
我们始终令最短的数组为n1,这样有利于我们下面的操作。
假设我们要查找在合并后数组中第k大的元素。
(1)我们分别在两个数组内查看K/2处的元素(如果K/2大于该数组的长度,便查询该数组末尾元素)。
(2)比较这两个元素的大小,假如n1[k/2]<n2[k/2],则说明n1数组k/2处的元素一定不是总的第k大的元素。因为我们假设,n2前k/2-1个元素都小于n1第k/2个元素,那么最大也只是合并后后第k-1个元素,所以我们将n1前k/2个元素全部删除(下标变了,并没有真正的删除)。并将K减去删除掉的元素个数。
重复以上步骤,直到k=1或某个数组被删除空。

只是分享了大概的思路,请理解大概之后,配合代码使用。
java代码

public double findMedianSortedArrays(int[] nums1, int[] nums2) {
		 int m = nums1.length;
		 int n = nums2.length;
		 return (helper(nums1,nums2,0,0,(m+n+1)/2)+helper(nums1, nums2,0,0, (m+n+2)/2))*0.5;
	 }
	 public int helper(int[] a,int[]b,int aS,int bS, int k) {
		 int aE = a.length-1;
		 int bE = b.length-1;
		 int len1 = aE-aS+1;
		 int len2 = bE-bS+1;
		 System.out.println("----------------");
		 System.out.println("   len1:"+len1+"    len2:"+len2+"  k:"+k);
		 if(len1>len2) {
			 return helper(b,a,bS,aS,k);
		 }
		 if(len1==0) {
			 return b[bS+k-1];
		 }
		 if(k == 1) {
			 return Math.min(a[aS], b[bS]);
		 }
		 int i = aS+Math.min(k/2, len1)-1;
		 int j = bS+Math.min(k/2, len2)-1;
		 System.out.println("i:"+i+"   j:"+j+"   len1:"+len1+"    len2:"+len2+"  k:"+k);
		 System.out.println("----------------");
		 if(a[i]>=b[j]) {
			 k -= (j-bS+1);
			 bS = j+1;
		 }
		 else {
			 k -= (i-aS+1);
			 aS = i+1;
		 }
		 return helper(b,a,bS,aS,k);
	 }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值