215 Kth Largest Element in an Array 快排

题目：在无序的数组中找到第k大的元素，也就是若长度为n的数组从小到大排列时，下标为n-k的元素。

注意Example2：第4大的元素是4，也就是数组中出现的两个5分别是第2大和第3大的数字。

解法一：直接利用sort函数排序后，取第k大的元素。

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

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

解法二：快排

时间复杂度是：O(n)

注意几个问题：

1）k的取值：是不变的啊！ 不会因为递归时在privot的左边或者右边寻找而改变，因为nums的长度n是不变的，元素的下标也是不变的；

2）partition函数递归调用自己时是要加return的！不然会报错。

class Solution {
public:

     int partition(vector<int>& nums, int k, int l, int r){
       int n = nums.size();
        swap(nums[l],nums[rand()%(r-l+)+l]);   //生成随机数
        int p = nums[l];

        //nums[l+1...i) <=v ; nums(j...r] >=v i和j 是需要考察的元素，所以是开区间
        int i = l+, j = r;     //初始定义区间为空
        while(true){
            while(i<=r && nums[i]<p)
                i++;
            while(j>=l+ && nums[j]>p)
                j--;
            if(i>j) break;
            swap(nums[i], nums[j]);
            i++;
            j--;
        }
        swap(nums[l], nums[j]);

        if(j == n-k) return nums[j];
        else if(j < n-k) return partition(nums, k, j+,r);
        else return partition(nums, k, l, j-);

    }

    int findKthLargest(vector<int>& nums, int k) {
        srand(time(NULL));
        int n = nums.size();
        return partition(nums, k, , n-);

    }

};

终于把快排思想的写对了，前一天晚上一直在报错，因为上面提到的两个点没有想明白，还是早上效率高 :)

解法二：思路是将比privot大的元素放到privot左边， 小的放到privot右边。

class Solution {
public:
    int quick_sort(vector<int>& nums, int k, int left, int right){
        swap(nums[left], nums[left + rand()%(right-left+)]);  //随机取在nums中取一个数 与 nums[left]交换位置
        int p = nums[left];
        int i = left+, j = right;
        while(){
            while( i<=right && nums[i] >= p)  // p = nums[left]
                i++;
            while( j>=left+ && nums[j] <= p)
                j--;
            if(i>=j)
                break;
            swap(nums[i], nums[j]);
            i++;
            j--;
        }
       // swap(nums[left], nums[j]);
        swap(p, nums[j]);    // 若交换p和nums[j] 因为还需要把 p 的值再赋给nums[left]
        nums[left] = p;

        if(j == k-)
            return nums[j];
        else if(k- > j)
            // k-1为4，j为3时
            // 第4大的数 < 第3大的数 数组是从大到小排序 故从右半部分寻
            return quick_sort(nums, k, j+, right);
        else
            return quick_sort(nums, k, left, j-);
    }    

    int findKthLargest(vector<int>& nums, int k) {
       return quick_sort(nums, k, , nums.size()-);
    }

};