LintCode-第k大元素

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

您在真实的面试中是否遇到过这个题？ 

Yes

样例

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

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

注意

你可以交换数组中的元素的位置

挑战

要求时间复杂度为O(n），空间复杂度为O(1）

标签 Expand  

相关题目 Expand 

分析：利用快排的思想，不断partition,

代码：

class Solution {public:    /*     * param k : description of k     * param nums : description of array and index 0 ~ n-1     * return: description of return     */    int kthLargestElement(int k, vector<int> nums) {        // write your code here        return findKthLargestElement(k,nums,0,nums.size()-1);    }    int findKthLargestElement(int k,vector<int>& nums,int start,int end)    {        if(start==end)            return nums[start];        int index = partition(nums,start,end);        if(end-index+1==k)            return nums[index];        else if(end-index+1>k)            return findKthLargestElement(k,nums,index+1,end);        else            return findKthLargestElement(k-(end-index+1),nums,start,index-1);    }    int partition(vector<int>&nums,int start,int end)    {        int x =  (start+end)/2;        swap(nums[start],nums[x]);        int i = start+1;        int j = i;        while(i<=end&&j<=end)        {            if(nums[j]<nums[start])            {                swap(nums[i],nums[j]);                i++;            }            j++;        }        swap(nums[start],nums[i-1]);        return i-1;    }};