215. Kth Largest Element in an Array

一、题目简述

给定一个未排序的序列，找到其第K大的元素。
示例：
给定[3,2,1,5,6,4]，k=2,返回5。
注意：
假定K总是有效的，即，$1\leq k\leq lens$
函数原型：int findKthLargest(vector<int>& nums, int k)

二、编程思路

本题的编程思路类似归并排序，对于一个序列$nums$，可以随机选择其中一个元素$val$，为方便考虑，我们选取序列中心元素作为$val$，将大于$val$的数字放入$lt\_vct(larger\ than\ vector)$中，等于的放入$eq\_vct(equal\ vector)$中，小于的放入$st\_vct(smaller\ than\ vector)$中。
则

vector长度	lt_vct_len	eq_vct_len	st_vct_eq
下标范围	$[0,lt\_vct\_len-1]$	$[lt\_vct\_len,lt\_vct\_len+eq\_vct\_len-1]$	$[lt\_vct\_len+eq\_vct\_len,lt\_vct\_len+eq\_vct\_len+st\_vct\_eq-1]$

• 若k落在eq_vct的范围内，说明$val$即是第K大元素，返回$val$；
• 若k落在lt_vct的范围内，说明第K大元素在lt_vct中，则在lt_vct中继续寻找；
• 若k落在st_vct的范围内，说明第K大元素是st_vct中的第K-lt_vct_len-eq_vct_len大元素；
  注意：数组下标范围及与K的关系。

三、代码设计、

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        return kthLargest(nums, k);
    }
    int kthLargest(vector<int>& nums, int k){
        int end = nums.size()-1;
        vector<int> larger, smaller,equal;
        int mid_idx = floor(end / 2);
        //cout << nums[mid_idx] << endl;
        for (int i = 0; i < nums.size(); i++){
            if (nums[i] > nums[mid_idx]){
                larger.push_back(nums[i]);
            }
            else if (nums[i] < nums[mid_idx]){
                smaller.push_back(nums[i]);
            }
            else{
                equal.push_back(nums[i]);
            }
        }
        int lt_size, st_size, eq_size;
        lt_size = larger.size();
        st_size = smaller.size();
        eq_size = equal.size();
        if ((k-1 >= lt_size) && (k-1 <= eq_size + lt_size-1)){
            //递归基
            //说明nums[mid_idx]为第K大的数
            return nums[mid_idx];
        }
            //递归步
        else if((k-1<=lt_size-1) && (k-1>=0)){
            return kthLargest(larger, k);
        }
        else if((k-1>=lt_size+eq_size)&& (k-1<=lt_size+eq_size+st_size-1)){
            return kthLargest(smaller, k - lt_size - eq_size);
        }
        else{
            cout << "bug here!" << endl;
            return -9999;
        }
    }
};

四、编程心得

数组要小心下标，递归要有递归基。