215. Kth Largest Element in an Array

一、题目简述

给定一个未排序的序列，找到其第K大的元素。
示例：
给定[3,2,1,5,6,4]，k=2,返回5。
注意：
假定K总是有效的，即，1≤k≤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;        }    }};

四、编程心得

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