C++用vector容器实现堆的相关算法

C++用vector容器实现堆的相关算法

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

int main() {
	vector<int> v = {0, 1, 2, 3, 4, 5, 6};
	make_heap(v.begin(), v.end()); // 以vector为底层容器构造大根堆(底层是完全二叉树)
	for (auto i : v) {
        cout << i << " "; // 6 4 5 3 1 0 2
    }
    cout << endl;
    v.push_back(7);
    push_heap(v.begin(), v.end());// 在vector尾部添加元素一直上溯到相应位置以满足大根堆性质
    for (auto i : v) {
		cout << i << " "; // 7 6 5 4 1 0 2 3
	}
    cout << endl;
    pop_heap(v.begin(), v.end());// 将头部元素弹出放到vector尾部进行相应的调整以满足大根堆的性质
    cout << v.back() << endl; // 7
    v.pop_back();// 弹出尾部元素
    for (auto i : v)
    {
        cout << i << " "; // 6 4 5 3 1 0 2
    }
    cout << endl;
    sort_heap(v.begin(), v.end());// 从小到大排序满足递增性质
    for (auto i : v)
    {
        cout << i << " "; // 0 1 2 3 4 5 6
    }
	return 0; 
}

运行结果