堆的创建,插入,删除

堆(Heap)可以视为完全二叉树

它的数据结构就是数组对象,在堆中,都是以下标进行访问的。

堆分为:

 最大堆:每个父节点的都大于孩子节点。

 最小堆:每个父节点的都小于孩子节点。

如果有组数据:int a [] = {10,11, 13, 12, 16, 18, 15, 17, 14, 19};

构建成大堆:


创建成小堆:


因为大堆和小堆代码大致差不多,所以可以进行复用,这里就用到了仿函数。

我用的代码是大堆,但是我想创建小堆时,直接将




代码:

Heap.h

#pragma once
#include <vector>

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

//大堆
template <class T, class Compare = Greater<T>>
class Heap
{
public:
	Heap()//无参构造函数
	{}
	Heap(T* a, size_t n)//构造函数
	{
		_a.reserve(n);
		for (size_t i = 0; i < n; i++)
		{
			_a.push_back(a[i]);
		}
		for (int i = (_a.size() - 2) / 2; i >= 0; i--)
		{
			_AdjustDown(i);
		}
	}
	void Push(const T& x)//插入数据
	{
		_a.push_back(x);
		_AdjustUp(_a.size() - 1);
	}
	void Pop()//删除数据
	{
		assert(!_a.empty());
		swap(_a[0], _a[_a.size() - 1]);
		_a.pop_back();
		_AdjustDown(0);
	}
	const T& Top() const
	{
		return _a[0];
	}
protected:
	void _AdjustDown(int root)//向下调整算法用于创建堆
	{
		Compare comFunc;
		int parent = root;
		int child = root * 2 + 1;
		while (child < _a.size())
		{
			if ((child+1<_a.size()) && comFunc(_a[child + 1],_a[child]))
			{
				++child;
			}
			//到这里了,在大堆里child就是最大的那个孩子,小堆相反
			if (child<_a.size() && comFunc(_a[child] ,_a[parent]))//大堆里,孩子比父亲大
			{
				swap(_a[child], _a[parent]);//交换
				parent = child;
				child = parent * 2 + 1;
			}
			else
			{
				break;
			}
		}
	}
	void _AdjustUp(int child)//向上调整算法用于插入数据
	{
		Compare comFunc;
		int parent = (child - 1) / 2;
		while (child > 0)
		{
			if (comFunc(_a[child] , _a[parent]))
			{
				swap(_a[child], _a[parent]);
				child = parent;
				parent = (child - 1) / 2;
			}
			else
			{
				break;
			}
		}
	}
protected:
	vector<T> _a;
};

void TestHeap()
{
	int a[] = { 10, 11, 13, 12, 16, 18, 15, 17, 14, 19 };
	Heap<int> hp1(a, sizeof(a) / sizeof(a[0]));
	hp1.Push(34);
	hp1.Pop();
}

Test.cpp

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

#include "Heap.h"

int main()
{
	TestHeap();
	return 0;
}




  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值