【数据结构】堆,堆实现优先级队列,堆排序

前端之家收集整理的这篇文章主要介绍了【数据结构】堆,堆实现优先级队列,堆排序前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

1.什么是堆?

堆是一种数据结构,底层是一种数组对象,它可以被视为一棵完全二叉树结构

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

2.堆数据结构二叉树存储。

如图所示是个大堆,只能保证父节点比孩子节点大。所以下标为0是整个堆最大的,但无法确定下标为1,2的数据哪个更大


3.堆数据结构和优先级队列的代码实现

思想:从第一个非孩子节点的下标开始向下调整,保证父节点大于子节点否则交换。(大堆),小堆与之相反,所以,这里用到仿函数和模板的模板参数,使得代码能够复用,根据传的参数决定是大堆还是小堆,此处就体现出了c++的优势。若是C的话,这里只能老老实实把相似的代码再写一遍。

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

template<class T>
struct Grearer
{
	bool operator()(const T& left,const T& right)
	{
		return left > right;
	}
};

template<class T>
struct Less
{
	bool operator()(const T& left,const T& right)
	{
		return left < right;
	}
};
template<class T,class Compare=Grearer<T>>    //模板的模板参数,默认值为大于
class Heap
{
public:
	Heap()
	{}
	Heap(T* a,size_t size)    //建堆
	{
		for (size_t i = 0; i < size; i++)  //数组内容拷贝给成员_a
		{
			_a.push_back(a[i]);
		}
		for (int i = (size - 2) / 2; i >= 0; i--)   //从最后一个非叶子节点向下调整
		{
			_AjustDown(i);
		}
	}

	void Push(const T& x)
	{
		_a.push_back(x);     //先插到堆的最后一个位置
		_AjustUp(_a.size() - 1);   //再对最后一个数据向上进行调整
	}

	void Pop()
	{
		assert(!_a.empty());    //对空堆进行断言
		swap(_a[0],_a[_a.size() - 1]);    //交换堆的第一个数据和最后一个
		_a.pop_back();                    //相当于删除了第一个数据
		_AjustDown(0);                     //对刚放在首位置的数据进行向下调整
	}

	size_t Size()           //堆节点个数
	{
		return _a.size();
	}

	const T& Top()     //堆的首个数据
	{
		return _a[0];
	}

	bool Empty()    //堆是否为空
	{
		return _a.empty();
	}
protected:
	void _AjustUp(int root)     //向上调整
	{
		size_t child = root;
		size_t parient = (child - 1) / 2;
		
		while (child > 0)
		{
			Compare com;
			//if (_a[child] > _a[parient])
			if (com(_a[child],_a[parient]))
			{
				swap(_a[child],_a[parient]);
			}
			else
			{
				break;
			}
			child = parient;
			parient = (child - 1) / 2;
		}

	}
	void _AjustDown(int root)   //向下调整
	{
		size_t parient = root;
		size_t child = 2 * parient + 1;
		
		while (child<_a.size())
		{
			Compare com;
			if ((child+1<_a.size())&&com(_a[child + 1],_a[child]))  //此处一定要先考虑到右孩子不存在的情况
			{
				++child;
			}
			//if (_a[parient] < _a[child])
			if (com(_a[child],_a[parient]))
			{
				swap(_a[parient],_a[child]);
			}
			parient = child;
			child = 2 * parient + 1;
		}
	}
private:
	vector<T>_a;
};

template<class T,class Compare=Grearer<T>>
class PriorityQueue    //优先级队列
{
public:
	void Push(const T&x)
	{
		_h.Push(x);
	}
	void Pop()
	{
		_h.Pop();
	}
	size_t Size()
	{
		return _h.Size();
	}
	const T& Top()
	{
		return _h.Top();
	}
	bool Empty()
	{
		return _h.Empty();
	}
private:
	Heap<T,Compare> _h;
};

void test()
{
	int a[10] = { 10,16,18,12,11,13,15,17,14,19 };
	Heap<int,Less<int>> hp(a,10);
	hp.Push(30);
	hp.Pop();
}
void TestPriority()  //测试优先级队列
{
	PriorityQueue<int> pq;
	pq.Push(3);
	pq.Push(2);
	pq.Push(1);
	pq.Push(9);
	cout << pq.Top() << endl;
	cout << pq.Size() << endl;
	pq.Pop();
<h2>}
</h2>

4.堆排序

为什么要使用堆排序呢,因为它的时间复杂度只有O(N*lgN),是一种比较好的排序算法。

以升序为例,基本思想:建好大堆后,把堆的第一个数据的堆的最后一个数据交换,则最后一个位置放置了整个堆中最大的数,此时除了下表为0的数据,剩下的满足大堆,再对剩下的堆进行向下调整(不包括最后一个数据),再把下标为0的数据和堆的倒数第二个数据进行交换,以此类推,最后得到一个有序的数组。
降序反之,建立小堆。算法类似。

升序的代码实现:

#pragma once
#include<iostream>

using namespace std;
void _AjustDown(int* a,size_t size,int root)
{
	size_t parient = root;
	size_t child = 2 * parient + 1;

	while (child<size)
	{

		if ((child + 1<size) && (a[child + 1]> a[child]))  //此处一定要先考虑到右孩子不存在的情况
		{
			++child;
		}
		if (a[parient] < a[child])
		{
			swap(a[parient],a[child]);
		}
		parient = child;
		child = 2 * parient + 1;
	}
}


	void HeapSort(int* a,size_t size)    
	{
		for (int i = (size - 2) / 2; i >= 0; i--)   //建好一个大堆
		{
			_AjustDown(a,size,i);
		}
		for (size_t i = 0; i < size; i++)
		{
			swap(a[0],a[size - 1-i]);    //把小标为0的数放在堆的倒数i+1结点处
			_AjustDown(a,size - 1 - i,0);   //调整除了最后一个数字的剩下的堆。
		}
		//return a;
	}

	

void TestHeapSort()
{
	int a[10] = { 10,19 };
	HeapSort(a,10);
}

5.堆的另一种应用——大数据运算

如1亿个数中找出最大的前100个数 (1亿内存放不下)。

类似的题就可以用堆来解决。从1亿数中取出100个数建立一个100数的小堆,从硬盘一个个读取数据,若读出的数比下标为0(堆里最小的数)大,则插入堆,否则丢弃,最后这100个数据的堆保存的就是这1亿个数里面最大的前100个。

猜你在找的数据结构相关文章