1921 Problem C 哈夫曼树

问题 C: 哈夫曼树

时间限制: 1 Sec  内存限制: 32 MB
提交: 76  解决: 44
 

题目描述

哈夫曼树，第一行输入一个数n，表示叶结点的个数。需要用这些叶结点生成哈夫曼树，根据哈夫曼树的概念，这些结点有权值，即weight，题目需要输出所有结点的值与权值的乘积之和。

输入

输入有多组数据。
每组第一行输入一个数n，接着输入n个叶节点（叶节点权值不超过100，2<=n<=1000）。

输出

输出权值。

样例输入

2
2 8 
3
5 11 30 

样例输出

10
62

经验总结

可以建哈夫曼树，也可以直接用优先队列实现，我这里为了练习哈弗曼树的自顶而下遍历方式，所以就用建树的方法啦~~

正确代码

#include <cstdio>
#include <climits>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn=1100;
struct HuffNode
{
	int w,parent,lchild,rchild,flag;
}Node[maxn*2];
void SearchMin(int &a,int &b,int n)
{
	int min=INT_MAX;
	for(int i=1;i<=n;++i)
	{
		if(Node[i].parent==0&&Node[i].w<min)
		{
			min=Node[i].w;
			a=i;
		}
	}
	min=INT_MAX;
	for(int i=1;i<=n;++i)
	{
		if(Node[i].parent==0&&Node[i].w<min&&i!=a)
		{
			min=Node[i].w;
			b=i;
		}
	}
	if(a>b)
	{
		swap(a,b);
	}
}
int HuffmanCode(int n,int * w)
{
	int m=2*n-1;
	for(int i=1;i<=n;++i)
	{
		Node[i].parent=Node[i].lchild=Node[i].rchild=0;
		Node[i].w=w[i];
	}
	for(int i=n+1;i<=m;++i)
	{
		int a,b;
		SearchMin(a,b,i-1);
		Node[i].lchild=a;
		Node[i].rchild=b;
		Node[i].w=Node[a].w+Node[b].w;
		Node[i].parent=0;
		Node[a].parent=Node[b].parent=i;
	}
	int ans=0,index=0;
	for(int i=1;i<=m;++i)
		Node[i].flag=0;
	while(m)
	{
		if(Node[m].flag==0)
		{
			Node[m].flag=1;
			if(Node[m].lchild!=0)
			{
				m=Node[m].lchild;
				++index;
			}
			else if(Node[m].rchild==0)
			{
				ans+=Node[m].w*index;
			}
		}
		else if(Node[m].flag==1)
		{
			Node[m].flag=2;
			if(Node[m].rchild!=0)
			{
				m=Node[m].rchild;
				++index;
			}
		}
		else
		{
			Node[m].flag=0;
			m=Node[m].parent;
			--index;
		}
	}
	return ans;
}
int main()
{
	int n,w[maxn];
	while(~scanf("%d",&n))
	{
		for(int i=1;i<=n;++i)
		{
			scanf("%d",&w[i]);
		}
		int ans=HuffmanCode(n,w);
		printf("%d\n",ans);
	}
    return 0;
}