寻找第K个大的数----编程题

标题：寻找第 K 大

有一个整数数组，请你根据快速排序的思路，找出数组中第K大的数。

给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间)，请返回第K大的数，保证答案存在。

测试样例：

[1,3,5,2,2],5,3

返回：2

分析 

1. 进行一次快排（将大的元素放在前半段，小的元素放在后半段）,假设得到的中轴为p

2. 判断 p - low + 1 == k ，如果成立，直接输出a[p]，（因为前半段有k - 1个大于a[p]的元素，故a[p]为第K大的元素）

3. 如果 p - low + 1 > k， 则第k大的元素在前半段，此时更新high = p - 1，继续进行步骤1

4. 如果p - low + 1 < k， 则第k大的元素在后半段， 此时更新low = p + 1, 且 k = k - (p - low + 1)，继续步骤1.

 代码实现

     /**
     * 第K大的数
     *
     */
    public static int findKth(int[] a, int n, int K) {
        if (a.length == 0) {
            return -1;
        }
        return findKth(a,0,n-1,K);
    }
    private static int findKth(int[] a, int low, int high, int K) {
        int par = partition(a,low,high);
        if (par - low + 1 == K) {
            return a[par];
        }else if (par - low + 1 > K) {
            return findKth(a,low,par - 1,K);
        }else {
            return findKth(a,par + 1,high,K - par + low - 1);
        }
    }

    private static int partition(int[] a, int left, int right) {
        int tmp = a[left];
        while (left < right) {
            while (left < right && a[right] >= tmp) {
                right--;
            }
            a[left] = a[right];
            while (left < right && a[left] <= tmp) {
                left++;
            }
            a[right] = a[left];
        }
        a[left] = tmp;
        return left;
    }


    //检验
    public static void main(String[] args) {
        int[] array = {1,3,5,2,2};
        System.out.println(findKth(array, 5, 3));    //2
    }