class Solution {
public:
int quickselect(vector<int>& nums,int l,int r,int k)
{
if(l==r) return nums[k];
题目里k小于数组长度,不断地在一个小区间里找数,最后缩小区间找的就剩这个数了,此时l==r
int part=nums[l];//划分元素,此时part是数组第一个元素,后面不断变换
int i=l-1,j=r+1;
while(i<j){
do{
i++;
}while(nums[i]<part);
do{
j--;
}while(nums[j]>part);
这样划分完i和j中间那个位置就是part,此时part在他正确的位置上
if(i<j) swap(nums[i],nums[j]);//j(包含j)坐标往后的都是大于的
}//直接比较元素的位置
if(k<=j) return quickselect(nums,l,j,k);左边位置
else return quickselect(nums,j+1,r,k);右边位置
}
int findKthLargest(vector<int>& nums, int k) {
int n=nums.size();
return quickselect(nums,0,n-1,n-k); }};
堆排序
class Solution {
public:
void maxheapify(vector<int>& a,int i ,int size){
int l=i*2+1,r=i*2+2,largest=i;父节点i的左孩子和右孩子分别为2i+1,2i+2
if(l<size&&a[l]>a[largest])
largest=l;
if(r<size&&a[r]>a[largest])
largest=r;
if(largest!=i)现在的largest是新的符合大根堆的父节点
{
swap(a[i],a[largest]);
maxheapify(a,largest,size);
对于这个父节点所更新的新位置,我们需要继续递归,看他的左右孩子是否符合大根堆}}
void buildmaxheap(vector<int>& a,int size){
int i;
for(i=size/2-1;i>=0;i--)
二叉树最后一个非叶子(父)节点的索引为 heapSize / 2 - 1,数组从0开始
{
maxheapify(a, i, size);
}}
int findKthLargest(vector<int>& nums, int k) {
int size=nums.size();
buildmaxheap(nums,size);
上面两个函数是建立完大根堆了,但是只能保证父节点大于子节点,还是找不出第k个大的
删除根节点,相当于删除了一个最大的大根堆删除元素是把末节点元素补充到堆顶,然后再调整
for(int i=nums.size()-1;i>=nums.size()-k+1;i--)
这里重新算这个nums.size(),是因为每次数组大小都变
{0是根节点,末尾的元素就是最末节点的元素
swap(nums[0],nums[i]);
size--;减掉删除的
maxheapify(nums, 0, size);根节点变了,从根节点开始重新调整 }
return nums[0]; }};