哈夫曼树

来源： 2010年北京邮电大学计算机研究生机试真题

#include "iostream"
#include "stdio.h"
#include "math.h"
#include "vector"
#include "queue"
#include "memory.h"
#include "algorithm"
#include "string"
using namespace std;
int n;

struct Tree
{
	struct Tree *l,*r;
	int x;
	bool leaf;
	bool used;
	Tree()
	{
		l=r=NULL;
		used=false;
	}
};

Tree *Q[10010];
int cnt;

int Dfs(Tree *r,int level)
{
	if(r->leaf)
		return r->x*level;
	return Dfs(r->l,level+1)+Dfs(r->r,level+1);
}

Tree* Select(bool *emptyflag)
{
	int minn=100*1001;//n<=100,每个数的权值<=1000,所以队列Q中每个节点的x<=100*1000
	int k=-1;
	for(int i=0;i<cnt;i++)
		if(!Q[i]->used&&Q[i]->x<minn)
		{
			minn=Q[i]->x;
			k=i;
		}
		if(k!=-1){
			Q[k]->used=true;
			return Q[k];}
		else{
			*emptyflag=true;
			return NULL;}
}

int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		int i,j,x;
		Tree *temp,*q1,*q2;
		cnt=0;
		for(i=0;i<n;i++)
		{
			scanf("%d",&x);
			q1=new Tree;
			q1->x=x;
			q1->leaf=true;
			Q[cnt++]=q1;
		}
		if(n==1)
		{
			printf("%d\n",x);
			continue;
		}

		bool empty=false;
		while(true)
		{
			q1=Select(&empty);
			q2=Select(&empty);
			if(empty)
				break;

			temp=new Tree;
			temp->l=q1;	
			temp->r=q2;
			temp->leaf=false;
			temp->x=q1->x+q2->x;
			Q[cnt++]=temp;
		}

		printf("%d\n",Dfs(q1,0));
	}
}

—————————————————————————————————

本文原创自Slience的csdn技术博客。

本博客所有原创文章请以链接形式注明出处。

欢迎关注本技术博客，本博客的文章会不定期更新。

大多数人想要改造这个世界，但却罕有人想改造自己。

世上没有绝望的处境，只有对处境绝望的人。

                                              ————By slience

—————————————————————————————————