网易2016实习 寻找第K大的数

题目：

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

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

测试样例：

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

返回：2

思路：采用快速排序的思想，每次根据base划分，根据K值判断在左边还是右边再次进行排序。

代码：

class Finder {
public:
    int findKth(vector<int> a, int n, int K) {
        // write code here
        helper(a,K,0,n-1);
        return a[K-1];
    }
    
    void helper(vector<int>& a, int K, int left, int right)
        {
        int base = a[left];
        int i = left, j = right;
        while(i < j)
            {
            while(i < j && a[j] < base)
                --j;
            a[i] = a[j];
            while(i < j && a[i] >= base)
                ++i;
            a[j] = a[i];
        }
        a[i] = base;
        if(K-1 < i)
            {
            helper(a,K,left,i-1);
        }
        else if(K-1 > i)
            {
            helper(a, K, i+1 ,right);
        }
        else
            return;
    }
};