lintcode5-5. 第k大元素

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

 注意事项

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

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

Yes

样例

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

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

挑战 

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

思想利用分治法进行处理：使用快速排序，和快速排序不同的是每次分为两个部分时仅需要考虑其中一个部分，这样就会使得时间复杂度由O(nlogn)降为O（n），这里我并没有使用迭代的写法，因为使用迭代时，数据量很大时内存会不够用。

class Solution {public:    /*     * @param n: An integer     * @param nums: An array     * @return: the Kth largest element     */     //快速排序    int quickSort(vector<int> &nums,int begin,int end){         int key=nums[begin];         int first=begin,last=end;                  while(first<last){             while(nums[last]>=key&&first<last){                last--;             }                 nums[first]=nums[last];             while(nums[first]<=key&&first<last){                first++;             }                 nums[last]=nums[first];         }         nums[first]=key;         return first;     }         int kthLargestElement(int n, vector<int> &nums) {        // write your code here        int length=nums.size();        int begin=0,end=length-1;                int flag1=quickSort(nums,begin,end);        while(1){            for(int i=0;i<length;i++)                cout<<nums[i]<<"  ";            cout<<endl;            if(length-flag1>n){                begin=flag1+1;                flag1=quickSort(nums,begin,end);            }            else if(length-flag1<n){                end=flag1-1;                flag1=quickSort(nums,begin,end);            }            else                 break;        }                    for(int i=0;i<length;i++)            cout<<nums[i]<<"  ";        return nums[length-n];    }};