C++ 堆排序和堆的其他基本操作

在刷到一道堆的题目的时候又忘记怎么写堆了，如果不写成优先队列的话。所以破竹写这篇博客了记录堆得相应的写法。

题目是leetcode215

首先heap的建立是：

void build_max_heap(vector<int>& nums){
        heap_size = nums.size();
        for(int i = (heap_size >> 1) - 1; i >= 0; -- i) max_heap(nums, i);
    }

这样子理解吧，除了根节点，从下往上每一个节点都要经过堆的维护。

堆的维护是：

int heap_size;
    void max_heap(vector<int>& nums, int id){
        int largest = id;
        int l = (id << 1) + 1, r = (id << 1) + 2;
        if(l < heap_size && nums[l] > nums[largest]) largest = l;
        if(r < heap_size && nums[r] > nums[largest]) largest = r;
        if(largest != id){
            swap(nums[largest], nums[id]);
            max_heap(nums, largest);
        }
    }

每次比较当前节点和子节点的大小，然后维护，维护结束之后要重新维护相应的改变了的子节点。

完整代码是：

class Solution {
private:
    int heap_size;
    void max_heap(vector<int>& nums, int id){
        int largest = id;
        int l = (id << 1) + 1, r = (id << 1) + 2;
        if(l < heap_size && nums[l] > nums[largest]) largest = l;
        if(r < heap_size && nums[r] > nums[largest]) largest = r;
        if(largest != id){
            swap(nums[largest], nums[id]);
            max_heap(nums, largest);
        }
    }
    void build_max_heap(vector<int>& nums){
        heap_size = nums.size();
        for(int i = (heap_size >> 1) - 1; i >= 0; -- i) max_heap(nums, i);
    }
public:
    int findKthLargest(vector<int>& nums, int k) {
        build_max_heap(nums);
        for(int i = 0; i < k; ++ i){
            swap(nums[0], nums[heap_size - 1]);
            -- heap_size;
            max_heap(nums, 0);
        }
        return nums[heap_size];
    }
};

这个会不断更新，如果遇到其他现在还没想到的操作的话。