LeetCode: Kth Largest Element in an Array

题目：
Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

For example,
Given [3,2,1,5,6,4] and k = 2, return 5.

Note:
You may assume k is always valid, 1 ≤ k ≤ array’s length.

解法一：
最直观的解法是将数组排序，再返回第k大的数。使用STL库中的sort函数，时间复杂度为nlog(n)。Accepted的代码如下：

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int n=nums.size();
        sort(nums.begin(),nums.end());
        return nums[n-k];
    }
};

代码的运行时间为26ms。

解法二：
分治算法：在数组中选取一个元素x（可选数组中的第一个元素），将数组中比x大的元素分到左边的数组中，其它元素分到右边的数组中，如果左边数组中的元素个数大于或等于k，则在左边数组中按相同方法找第k大的元素；若左边数组中的元素恰好等于k-1，则返回元素x；若左边数组中的元素个数小于k，设为n，则在右边数组中找第k-n-1大的元素。可以用递归来实现，算法复杂度为o(nlogn)。代码：

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int n=nums.size();
        if(n==1) return nums[0];

        vector<int> left;
        vector<int> right;
        for(int i=1;i<n;i++)
        {
            if(nums[i]>nums[0]) left.push_back(nums[i]);
            else right.push_back(nums[i]);
        }

        if(left.size()>=k) return findKthLargest(left,k);
        else if(left.size()==k-1) return nums[n/2];
        else return findKthLargest(right,k-left.size()-1);
    }
};

但是代码提交后超时了，选取的比较元素x应该是一个随机数，如果都指定为nums[0]的话，对倒序排列足够大的数组，可能会超时。
修改后Accepted的代码：

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int n=nums.size();
        if(n==1) return nums[0];

        int index=rand()%n;
        vector<int> left;
        vector<int> right;
        for(int i=0;i<n;i++)
        {
            if(i==index) continue;
            if(nums[i]>nums[index]) left.push_back(nums[i]);
            else right.push_back(nums[i]);
        }

        if(left.size()>=k) return findKthLargest(left,k);
        else if(left.size()==k-1) return nums[index];
        else return findKthLargest(right,k-left.size()-1);
    }
};

运行时间为9ms，算法效率提高。