leetcode——finding Median of Two Sorted Arrays题目分析

leetcode上面有道“Median of Two Sorted Arrays”关于求中位数的题目，时间复杂度要求为O(log (m+n))，因此正好在本文梳理下关于

求中位数的方法。

规则：

    1）元素个数为奇数：中间那个

    2）元素个数为偶数：中间两个数求平均数

A.首先，寻找中位数最简单直接的办法肯定是循环从头加起，而此种方法的时间复杂度显然为O(m+n)

public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
		int k=nums1.length+nums2.length;
		if(k%2==1){
			return findkth(nums1,nums2,k/2+1);
		}
		else{
			return (findkth(nums1,nums2,k/2)+findkth(nums1,nums2,k/2+1))/2;
		}
	}
	public static double findkth(int[] A,int[] B,int k){
		int i=0,j=0,n=0;
		while((i<A.length)&&(j<B.length)){
			if(A[i]<B[j]){
				n++;
				if(n==k){
					return A[i];
				}
				i++;
			}
			else{
				n++;
				if(n==k){
					return B[j];
				}
				j++;
			}
		}
		if(i<A.length){
			return A[i-1+k-n];
		}
		else{
			return B[j-1+k-n];
		}
	}

B.由于时间复杂度的要求，本问题很容易想到使用二分法解决，我们首先借助寻找第k大元素的方法解决。

二分法寻找第k大元素的两个核心点：

    1）如果数组A的元素个数小于k/2，则第k个元素不可能出现在数组B的前k/2个元素中，也就是说我们可以将数组B的前key/2个元素

舍去

    2）如果数组A的第k/2个元素小于数组B的第k/2个元素，则可以舍去数组A的前k/2个元素

即每次减少两数组任一数组前k-1个元素中的k/2个元素，直到k减少为1，或者两个数组中只有一个数组还有元素，另一个数组为空（即

B_start>=B.length）

public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
		int len=nums1.length+nums2.length;
		if(len%2==1){
			return findkth(nums1,0,nums2,0,len/2+1);
		}
		return (findkth(nums1,0,nums2,0,len/2)+findkth(nums1,0,nums2,0,len/2+1))/2;
	}
	public static double findkth(int[] A,int A_start,int[] B,int B_start,int k){
		if(A_start>=A.length){//此处==也可以，因为导致越界的原因在于当某一数组去除掉其中的k/2个元素后正好为空

			return B[B_start+k-1];
		}
		if(B_start>=B.length){
			return A[A_start+k-1];
		}
		if(k==1){
			return Math.min(A[A_start],B[B_start]);
		}
		int A_mid=A_start+k/2-1<A.length
		?A[A_start+k/2-1]
		:Integer.MAX_VALUE;
		int B_mid=B_start+k/2-1<B.length
				?B[B_start+k/2-1]
				:Integer.MAX_VALUE;		
		//如果数组A的元素个数小于k/2,将数组B的前key/2个元素舍去


		if(A_mid<B_mid){
			return findkth(A,A_start+k/2,B,B_start,k-k/2);//注意：k-k/2不等于k/2
		}
		else{
			return findkth(A,A_start,B,B_start+k/2,k-k/2);
		}
	}