寻找第K大——Java实现

题目说明：
有一个整数数组，请你根据快速排序的思路，找出数组中第K大的数。
给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间)，请返回第K大的数，保证答案存在。

测试样例：
[1,3,5,2,2],5,3
返回：2

Java代码：

public class Four {
	public static void quick_sort(int[] array,int start,int end){
		int low=start,high=end;
		int target=array[low];
		if(low < high){
		while (low < high) {
		    //从右向左找比基数小的
			while (low < high&&array[high] > target) {
				high--;
			}
			//填坑
			if (low < high) {
				array[low] = array[high];
				low++;
			}

            //从左往右找比基数大的
			while (low < high&&array[low] < target) {
				low++;
			}
			//填坑
			if (low < high) {
				array[high] = array[low];
				high--;
			}

		}

		//循环结束后，low和high指向同一个位置，将target放在该位置处
		array[low] = target;
		quick_sort(array, start, low - 1);
		quick_sort(array, low + 1, end);
		}
	}	

	public static int find_k(int k,int[] arr,int n){
		if(arr==null||n==0){
			return -1;
		}
		quick_sort(arr,0,n-1);
		return arr[k-1];
	}
	
	public static void main(String[] args) {
		int[] arr={1,3,5,2,2};
		int n = arr.length;
		int k = 3;
		System.out.println(find_k(k,arr,n));
	}
 
}

运行结果: