堆的实现

堆的实现

堆的介绍：

二叉堆的底层数据结构为数组，它可以被视为完全二叉树结构；

堆的特性：

1、父结点总是大于（或小于）子结点

2、每个节点的左子树和右子树都是一个最大堆（或最小堆）

如图：

堆的代码实现如下：

提示：本程序利用 仿函数与 模板实现的堆，大堆与小堆的代码只是一个大于小于号的问题，所以我通过模板传参，实现大小堆，代码复用了起来

<span style="font-size:18px;">#pragma once

#include<iostream>
#include<assert.h>
using namespace std;
#include<vector>

//仿函数

template <class T>
struct GreaterCompare
{
	bool operator()(const T& l, const T& r)
	{
		return l > r;
	}
};

template<class T>
struct LessCompare
{
	bool operator()(const T& l, const T& r)
	{
		return l < r;
	}
};

template <class T,class Compare = GreaterCompare<T>>
class Heap
{
public:
	Heap()
	{}

	Heap(const T* arr, int len)
	{
		assert(arr);
		assert(len > 0);
		int i = 0;
		//传入数据
		for (; i < len; ++i)
		{
			_arr.push_back(arr[i]);
		}
		//建堆
		i = _arr.size();
		for (i = (i - 2) / 2; i >= 0;--i)
		{
			_AdjustDown(i);
		}
	}

	int Size()
	{
		return _arr.size();
	}

	bool Empty()
	{
		return _arr.empty();
	}

	//插入数据
	void Push(const T& x)
	{
		_arr.push_back(x);
		_AdjustUp(_arr.size()-1);
	}

	//删除数据
	void Pop()
	{
		if (!Empty())
		{
			swap(_arr[0], _arr[_arr.size() - 1]);
			_arr.pop_back();
			_AdjustDown(0);
		}
	}

	T Top()
	{
		return _arr[0];
	}

	~Heap()
	{}
protected:
	void _AdjustDown(int root)
	{
		Compare com;
		int child = root * 2 + 1;
		while (child < (int)_arr.size())
		{
			if (child + 1 < (int)_arr.size() && com(_arr[child + 1], _arr[child]))
				++child;
			if (com(_arr[child] , _arr[root]))
			{
				swap(_arr[child], _arr[root]);
				root = child;
				child = 2 * root + 1;
			}
			else
			{
				break;
			}
		}
	}

	void _AdjustUp(int child)
	{
		Compare com;
		int root = (child - 1) / 2;
		while (child > 0)
		{
			if (com(_arr[child], _arr[root]))
			{
				swap(_arr[child], _arr[root]);
				child = root;
				root = (child - 1) / 2;
			}
			else
			{
				break;
			}
		}
	}
private:
	vector<T> _arr;
};

void TestHeap()
{
	int a[] = { 10, 11, 13, 12, 16, 18, 15, 17, 14, 19 };
	//默认是大堆
	Heap<int> hp(a, sizeof(a) / sizeof(a[0]));
	//传参小堆的调用
	//Heap<int,LessCompare<int>> hp(a, sizeof(a) / sizeof(a[0]));

	hp.Push(20);
	hp.Pop();
}</span>