寻找无序数组的第k大元素

问题：给定一个无序数组，找出该无序数组的第k大元素。

思路：1.先排序，那么第k个元素自然就是第k大的元素了。

2.使用最小堆，先利用前k个元素建立最小堆，再往后比较，如果有比堆顶元素大的，则调整最小堆，最后堆顶元素便是要找的值。

 

先看第一种解法，使用排序的方法，这里快速排序，代码如下：

	//方法1：先排序，第k个就是第k大元素
	public static int findKthNumber(int[] a,int k) {
		quickSort(a,0,a.length-1);
		return a[a.length-k];
	}
	
	//快速排序
	public static void quickSort(int[] a,int begin,int end) {
		if(begin>=end) {
			return;
		}
		
		int i=begin,j=end;
		int flag=a[begin];
		
		while(i<j) {
			//从右边扫描
			while(i<j && a[j]>flag) {
				j--;
			}
			//交换
			if(i<j) {
				int t=a[i];
				a[i]=a[j];
				a[j]=t;
				i++;
			}
			//从左边扫描
			while(i<j && a[i]<flag) {
				i++;
			}
			
			//交换
			if(i<j) {
				int t=a[i];
				a[i]=a[j];
				a[j]=t;
				j--;
			}
		}
		a[i]=flag;	
		//递归排序
		quickSort(a,begin,i-1);
		quickSort(a,i+1,end);
	}
	

再看第二种方法，使用最小堆：

	//方法2：利用最小堆
	public static int findKthNumber2(int[] a,int k) {
		//将前k个元素调整为最小堆
		buildMinHeap(a,k);
		for(int i=k;i<a.length;i++) {
			if(a[i]>a[0]) {
				a[0]=a[i];
				fixDown(a,0,k);
			}
		}
		return a[0];
	}
	/**
	 * @param a  待调整的堆
	 * @param len  堆的大小
	 */
	public static void buildMinHeap(int[] a,int len) {
		//从最后一个非叶子节点开始，向上调整
		for(int i=(len-2)/2;i>=0;i--) {
			fixDown(a,i,len);
		}
	}
	
	/**
	 * @param a  待调整的堆
	 * @param i  根节点
	 * @param len  堆的大小
	 */
	public static void fixDown(int[] a,int i,int len) {
		int tmp=a[i];
		int parent=i;
		int child=i*2+1;
		//如果还有左孩子
		for(;parent*2+1<len;parent=child) {
			child=parent*2+1;  //左孩子
			//如果存在右孩子，并且右孩子更小，则child指向右孩子
			if(child+1<len && a[child+1]<a[child]) {
				child++;
			}
			//如果父节点更小，则退出
			if(a[child]>a[parent]) {
				break;
			}else {
				int t=a[child];
				a[child]=a[parent];
				a[parent]=t;
			}
		}
		a[parent]=tmp;
	}