堆的实现

#include <iostream>
#include<time.h>
#include<stdio.h>
#include<stdlib.h>
#define MAX 10000
using namespace std; 
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
int heap[MAX],sz=0;
void push(int x)
{
	int i=sz++;
	while(i>0)
	{
		int p=(i-1)/2;//父节点 
		if(heap[p]<=x)
		{
			break;
		}
		heap[i]=heap[p];//父节点下移 
		i=p; //自己提上去 
	}
	heap[i]=x;
}
int pop()
{
	int ret=heap[0];//删除 
	int x=heap[--sz];//最大的数 
	int i=0;
	while(i*2+1<sz)
	{
		int a=i*2+1,b=i*2+2;//相对根节点的两个节点 
		if(b<sz&&heap[b]<heap[a])
		{
			a=b;
		}
		if(heap[a]>=x)
		{
			break;
		}
		heap[i]=heap[a];
		i=a;
	}
	heap[i]=x;
	return ret;
}
int main(void)
{
	srand(unsigned(time(NULL)));
	for(int i=0;i<10;i++)
	{
		int a=rand()%10;
		push(a);
		cout<<a<<" ";
	}
	cout<<endl;
	for(int i=0;i<10;i++)
	{
		int b=pop();
		cout<<b<<" ";
	}
	cout<<endl;
	return 0;
} 

阅读更多
文章标签: priority queue
个人分类: priority queue
上一篇0-1背包问题
下一篇并查集的实现
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭