5. 第k大元素

在数组中找到第k大的元素

样例
给出数组 [9,3,2,4,8]，第三大的元素是 4

给出数组 [1,2,3,4,5]，第一大的元素是 5，第二大的元素是 4，第三大的元素是 3，以此类推

挑战
要求时间复杂度为O(n)，空间复杂度为O(1)

注意事项
你可以交换数组中的元素的位置

这道题我用的算法书上的方法，原理跟快排的原理一致，分成3步

1. 确定一个基准元素
2. 以这个基准元素为分界，比他大的元素放在他的左边，比他小的元素放在他的右边，即将数组逆序排序
3. 确定划分的下标，若k比下标小则在左半边数组找，若比下标大则在右半边数组找

代码如下

int Random(int p,int r)
    {
        return p+rand()%r;//將随机数控制在p~(r-1)之间
    }
    int Partition(vector<int> &a,int p,int r)//以子数组的第一个元素为基准划分函数，返回划分位置
    {
        int i=p,j=r+1;
        int x=a[p];
        int t;
        while(true)
        {
            while(a[++i]>x&&i<r);//找到大于x的元素
            while(a[--j]<x);//找到小于x的元素
            if(i>=j) break;
            //若i<j，则表明a[i]<x,a[j]>x时a[i]在左边，a[j]在右边，
            //不符合划分的原则（大于x的在左边，小于x的在右边），
            //因此交换a[i]与a[j]
            t=a[i];
            a[i]=a[j];
            a[j]=t;
            //Swap(a[i],a[j]);
        }
        //j为划分点，将a[p]放到划分点的位置
        a[p]=a[j];
        a[j]=x;
        return j;
    }
    int RandomizedPartition(vector<int> &a ,int p,int r)//在数组中随机找到一个元素作为划分的基准元素
    {
        int i=Random(p,r);
    
        int t;
        t=a[i];
        a[i]=a[p];
        a[p]=t;
    
        return Partition(a,p,r);
    }
    int RandomizedSelect(vector<int> &a,int p,int r,int k)//分治找到第k个元素，跟快排的处理方式一样
    {
        if(p==r) return a[p];
        int i=Partition(a,p,r),
            j=i-p+1;//j为划分左半边数组的元素个数
        if(k<=j) return RandomizedSelect(a,p,i,k);//若左半边数组元素个数大于j，则继续在左边数组中找
        else return RandomizedSelect(a,i+1,r,k-j);//若左半边数组元素个数小于j，则在右边数组中找
    }
    int kthLargestElement(int n, vector<int> &nums) {
        return RandomizedSelect(nums,0,nums.size()-1,n);
        // write your code here
    }

在这里RandomizedPartition(vector &a ,int p,int r)函数我没有用上，不知道为什么随机找基准元素提交时某些测试数据会出错。如果有大佬知道的可以告诉我为什么

算法的时间复杂度为
平均为O(n)
最坏情况（即每次都在最小的元素处划分）为O(n^2)
算法的空间复杂度为
O(1)