leetcode.4.寻找两个数组的中位数

最新推荐文章于 2022-04-27 16:34:58 发布

学术大new牛

最新推荐文章于 2022-04-27 16:34:58 发布

阅读量71

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_36744209/article/details/104033881

版权

leetcode 专栏收录该内容

12 篇文章 0 订阅

订阅专栏

class Solution {
    	public double findMedianSortedArrays(int[] nums1,int[] nums2) {
		int n = nums1.length;
		int m = nums2.length;
		int left = (n + m + 1) / 2;
		int right = (n + m + 2) / 2;
		//将偶数和奇数的情况合并，如果是奇数，会求两次同样的k
		return (getKth(nums1,0,n - 1,nums2,0,m-1,left)) * 0.5;
		
	}
	private int getKth(int[] nums1,int start1,int end1,int[] nums2,int start2,int end2,int k) {
		int len1 = end1 - start1 + 1;
		int len2 = end2 - start2 + 1;
		//让len1的长度小于len2，这样就能保证如果有数组空了，一定是len1
		if(len1 > len2) return getKth(nums2,start2,end2,nums1,start1,end1,k);
		if(len1 == 0) return nums2[start2+k-1];
		
		if(k == 1) return Math.min(nums1[start1], nums2[start2]);
		
		int i = start1 + Math.min(len1, k / 2) - 1;
		int j = start2 + Math.min(len2, k / 2) - 1;
		
		if(nums1[i] > nums2[j]) {
			return getKth(nums1,start1,end1,nums2,j+1,end2,k-(j-start2+1));
		}
		else {
			return getKth(nums1,i + 1,end1,nums2,start2,end2,k-(i-start1+1));
		}
	}
}