PTA算法题：哈夫曼树

哈夫曼树，第一行输入一个数n，表示叶结点的个数。

需要用这些叶结点生成哈夫曼树，根据哈夫曼树的概念，这些结点有权值，即weight，题目需要输出哈夫曼树的带权路径长度（WPL）。

输入格式:

第一行输入一个数n，第二行输入n个叶结点（叶结点权值不超过1000，2<=n<=1000）。

输出格式:

在一行中输出WPL值。

输入样例:

5
1 2 2 5 9

输出样例:

37

#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;
}