寻找第K大

本题目来源于牛客网：https://www.nowcoder.com/questionTerminal/e016ad9b7f0b45048c58a9f27ba618bf

题目描述：

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

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

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

解题思路：
1.进行一次快排（partition)，讲数字大的放在前半段，小的放在后半段，假设得到的中轴位置为p
2.判断p-left+1与k的关系
1）p-left+1==k,直接输出a[p]
2) p-left+1>k,在前半段开始找，此时right=p-1,继续进行第1步
3）p-left+1<k,在后半段开始找，此时left=p+1,k=k-(p-left+1),继续进行步骤1

代码如下：

  import java.util.*;

public class Finder {
    public int findKth(int[] a, int n, int K) {
        return findKth(a,0,n-1,K);
    }
    public int findKth(int[] a,int left,int right,int k){
        int part=partition(a,left,right);
        if(k==part-left+1)  {
            return a[part];
        } else if(k>part-left+1){
            return findKth(a,part+1,right,k-part+left-1);
        }else{
        return findKth(a,left,part-1,k);
        }
    }
    public int partition(int[] a,int left,int right){
        int b=a[left];
        while(left<right){
            while(left<right&&a[right]<=b){
                right--;
            }
                a[left]=a[right];
            
            while(left<right&&a[left]>=b){
                left++;
            }
                a[right]=a[left];
        }
        a[left]=b;
        return left;
    }
}