【名企中等】寻找第K大

https://www.nowcoder.com/practice/e016ad9b7f0b45048c58a9f27ba618bf?tpId=117&&tqId=35010&rp=1&ru=/ta/job-code-high&qru=/ta/job-code-high/question-ranking

---

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

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

---

思路：
题面已经提示利用快排思想，快排的每一次查找都会把元素放到自己的最终位置，由此得出解题方法。 需要注意这里的第K大，第1大是数组的最后一个。

---

   public int findKth(int[] a, int n, int K) {
        // write code here
        return helper(a,a[0],0,a.length-1,K);
    }

    private int helper(int[] a, int flag, int start, int end, int K){
        int i = start;
        int j = end;
        while(i < j){
            while(j >= start && a[j] >= flag && i < j){
                j--;
            }
            if(i < j){
                swap(a, i, j);
            }
            while(i <= end && a[i] <= flag && i < j){
                i++;
            }
            if(i < j){
                swap(a, i, j);
            }

        }
        if(i < K){
            return helper(a,a[i+1],i+1,a.length-1,K);
        } else if(i > K){
            return helper(a,a[0],0,i-1,K);
        }else{
            return a[i];
        }
    }

    private void swap(int[] a, int i, int j){
        int temp = a[i];
        a[i] = a[j];
        a[j] = temp;
    }

快排的部分可以写得再精简些

int flag = A[start];
        int i = start+1;
        int j = end;
        while(i<=j && i <= end && j >= start){
            while (i<=end && A[i] < flag){
                i++;
            }
            while(j >= start && A[j] > flag){
                j--;
            }
            if(i < j){
                swap(A,i,j);
            } else{
                swap(A,start,j);
                break;
            }
        }