堆的实现和STL

#include "pch.h"
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;

const int maxn = 100;
int heap[maxn], n = 10;//heap为堆 n为元素个数

//堆heap数组在[low,high]范围进行向下调整，即与孩子比
//low为欲调整节点的数组下标，high为堆最后一个元素的下标
void downAdjust(int low, int high) {
	int i = low, j = i * 2;//i为欲调整节点，j为其左孩子
	while (j <= high) {//存在孩子节点
		//如果右孩子存在，且右孩子值大于左孩子
		if (j + 1 <= high && heap[j + 1] > heap[j]) {
			j = j + 1;//让j存储最大的孩子下标
		}
		//如果孩子中权值最大的权值比欲调整节点大
		if (heap[j] > heap[i]) {
			swap(heap[j], heap[i]);//交换最大权值的孩子与欲调整节点
			i = j;
			j = i * 2;
		}
		else {
			break;//孩子的权值均小于i，调整结束
		}
	}
}
/*
建堆：若序列元素个数为n，完全二叉树叶子节点个数为n/2+1 所以下标在
[1,n/2]范围内的都是非叶子节点。考研从n/2开始倒着枚举节点，随每个遍历到的
节点i进行[i,n]范围调整。
*/
void createHeap() {
	for (int i = n / 2; i >= 1; i--) {
		downAdjust(i, n);
	}
}

//删除堆顶元素
void deleteTop() {
	heap[1] = heap[n--];
	downAdjust(1, n);
}
//对heap数组在[low,high]范围进行向上调整
//对low一般设置为1，high表示欲调整节点的数组下标
void upAdjust(int low, int high) {
	int i = high, j = i /2;//i为欲调整节点，j为父节点
	while (j >= low) {
		if (heap[j] < heap[i]) {
			swap(heap[i], heap[j]);
			i = j;
			j = i / 2;
		}
		else {
			break;//父节点权值比欲调整节点i的权值大，调整结束
		}
	}
}
//添加元素
void insert(int x) {
	heap[++n] = x;//让元素个数加一，然后将数组末尾赋值为x
	upAdjust(1, n);
}

//堆排序
void heapSort() {
	createHeap();
	for (int i = n; i > 1; i--) {//倒着枚举，直到堆中只有一个元素
		swap(heap[i], heap[1]);//交换heap[i]与堆顶
		downAdjust(1, i - 1);//调整堆顶
	}
}

//STL <algorithm>库中
/*
make_heap():建堆
push_heap():在堆中添加元素。添加数到数组末尾然后调用
pop_heap():在堆中删除元素。调用之后pop_back数组最后一个数
sort_heap()：堆排序
*/
void printHep(vector<int> &v) {
	for (vector<int>::iterator it = v.begin(); it != v.end(); it++) {
		cout << *it << " ";
	}
	cout << endl;
}

int main() {
	vector<int> min = { 10,30,22,6,15,9 };
	make_heap(min.begin(), min.end(), greater<int>());//建立小根堆
	printHep(min);

	//插入元素
	min.push_back(20);
	push_heap(min.begin(), min.end(), greater<int>());

	//删除堆顶
	pop_heap(min.begin(), min.end(), greater<int>());
	min.pop_back();

	//堆排序
	sort_heap(min.begin(), min.end(), greater<int>());
	printHep(min);
	return 0;
}