Bootstrap

求数组中第K大的数

使用快排,第一趟快排过后,右边的元素都比枢轴大,左边的都比枢轴小。当我们要求第K大的数,只需要左边的元素个数是len-k,len是数组总长度。

第一次快排下来,设枢轴位置是mid,如果mid>k,说明k大元素应该在左边序列中,递归左边序列。当mid<k时,说明k元素在右边序列,递归右边序列,但是因为左边序列有mid个元素,所以只要求右边序列中k-mid个数即可。

程序中的思想就是,当这个数左边有len-k个元素时,这个元素即是k大元素。

程序代码如下:

public class KMax {
	public int partition(int[] a,int low,int high){
		int tmp=a[low];
		while(low<high){
			while(low<high && a[high]>tmp)
				--high;
			a[low]=a[high];
			while(low<high && a[low]<tmp)
				++low;
			a[high]=a[low];
		}
		a[low]=tmp;
		return low;
	}
	//这里的k意思是,左边序列的元素个数,左边有len-k个元素,此元素就是k大元素
	public int kMaxSort(int[] a,int low,int high,int k){
		int kmax;
		if(k<0)
			return 0;
		if(low>=high)
			return a[low];
		else{
			int mid=partition(a, low, high);
			//递归左边序列
			if(mid>k)
				kmax=kMaxSort(a, low, mid-1, k);
			//递归右边序列,但是k要变成k-mid
			else if(mid<k)
				kmax=kMaxSort(a, mid+1, high, k-mid);		
			else
				kmax= a[mid];
		}
		return kmax;
	}
	public static void main(String[] args) {
		KMax ktt=new KMax();
		int[] a={10,7,8,6,3,1,5,2,4,9};
		int k =10;
		int len = a.length;
		//注意k为len-k
 		int kmax=ktt.kMaxSort(a,0, len-1, len-k);
 		System.out.println(kmax);
	}
}

;