小顶堆

关注

原创最新推荐文章于 2025-07-05 09:02:54 发布 938 阅读 AI 写同款· GEO 优化 ›

小顶堆、小根堆 Small Heap 小顶堆。可以用vector实现。求最大的前k个数就可以构造一个k个元素的小顶堆，堆顶是k个大数中最小的，新元素比堆顶大就替换堆顶、排序，否则不管特征：父亲的值 < 儿子的值，元素从上到下从左到右紧凑排列（完全二叉树）过程： heapify：对当前节点不断向下交换直到没有大小颠倒，交换的时候优先选数值小的儿子换入堆：插在堆的末尾（数组的最后一个），然后不断【向上提升直到没有大小颠倒... 阅读详情

#ifndef HEAP_H__

#define HEAP_H__
#include <iostream>
#include <cassert>
#endif


template <class T>
class HEAP
{
public:
	HEAP(T& array, int size);
	HEAP() = default;
	~HEAP() { delete Heap; };
	void Init() const;
	HEAP<T>& Push(const T& elem);
	HEAP<T>& Pop(T& elem);
	void Print() {
		for(int i=1; i <= CurrentSize; i++)
			std::cout << Heap[i] <<", ";
	}
private:
	int MaxSize = 1000, CurrentSize = 0;
	T* Heap;
};


template <class T>
HEAP<T>::HEAP(T& array, int size) {
	CurrentSize = size;
	Heap = &array;
	Heap --;
}


template <class T>
void HEAP<T>::Init() const {
	assert(CurrentSize < MaxSize);
	T temp;
	int child;
	for(int parent=CurrentSize/2; parent >= 1; parent--){
		int nowindex = parent;
		while(nowindex >= 1) {
			child = 2*parent;
			if(child <CurrentSize && Heap[child+1] < Heap[child])
				child++;
			if( Heap[child] < Heap[nowindex]) {
				temp = Heap[child];
				Heap[child] = Heap[nowindex];
				Heap[nowindex] = temp;
			}
            else
                break;
			nowindex--;
		}
	}
}



int main(int argc, char const *argv[])
{
	int a[] = {10,100, 9, 8, 7, 5, 6, 3, 2, 4, 1};
	HEAP<int> heap(*a, 11);
	heap.Print();
	return 0;
}