215-数组中的第K个最大元素

题目

数组中的第K个最大元素
在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:

你可以假设 k 总是有效的，且 1 ≤ k ≤ 数组的长度。

我的思路

1、使用STL对原数组进行排序
2、自己实现快速排序，对数组进行排序

我的代码

//
// Created by HINTS on 2018/12/19.
//

#include <iostream>
#include <vector>
#include <c++/algorithm>

using  namespace std;

//第一种解法：使用STL
int findKthLargest1(vector<int>& nums, int k) {
    sort(nums.begin(), nums.end());
    return nums[nums.size() - k];
}

//第二种解法：自己实现快速排序
//返回p, 使得nums[l...p-1] < nums[p] ; nums[p+1...r] > nums[p]
int __partition(vector<int> &nums, int l , int r){
    int v = nums[l];
    int j = l;
    for (int i = l + 1; i <= r; ++i) {
        if(nums[i] < v){
            j++;
            swap(nums[j], nums[i]);
        }
    }
    swap(nums[l], nums[j]);
    return j;
}

void __quickSort(vector<int> &nums, int l, int r){
    if(l >= r)
        return ;
    int p = __partition(nums, l, r);
    __quickSort(nums, l, p-1);
    __quickSort(nums, p+1, r);
}
void quickSort(vector<int> &nums, int n){
    __quickSort(nums,0 , n-1);
}

int findKthLargest(vector<int>& nums, int k) {
    quickSort(nums, nums.size());
    return nums[nums.size() - k];
}
int main(){
    vector<int> nums = {3,2,3,1,2,4,5,5,6};
    quickSort(nums, nums.size());
    for (int i = 0; i < nums.size(); ++i) {
        cout << nums[i] <<endl;
    }
    //cout << findKthLargest(nums, 2)<<endl;
    return 0;
}