cjweffort

博客园 首页 联系 订阅 管理

http://ac.jobdu.com/problem.php?cid=1040&pid=29

题目描述:

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

输入:

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

输出:

输出权值。

样例输入:
5  
1 2 2 5 9
样例输出:
37
// 题目30:哈夫曼树.cpp: 主项目文件。
//不使用优先队列(堆结构)
#include "stdafx.h"
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <set>
#include <map>
#include <queue>
#include <stack>
using namespace std;


const int MAX=1<<30;
const int N=20003;
int a[N];
//int level[N];
bool used[N];

int n;

int selectMin(int length)
{
	int minIndex=-1,min=MAX;
	for(int i=0;i<length;i++)
	{
		if(!used[i])
		{
			if(a[i]<min)
			{
				min=a[i];
				minIndex=i;
			}
		}

	}
	return minIndex;
}

int huffmanTree()
{
	int wholeWeight=0;
	//memset(level,0,sizeof(level));
	memset(used,0,sizeof(used));
	for(int i=0;i<n-1;i++)
	{
		int index1=selectMin(n+i);
		used[index1]=true;
		int index2=selectMin(n+i);
		used[index2]=true;
		a[n+i]=a[index1]+a[index2];
		wholeWeight+=a[n+i];
	}
	return wholeWeight;
}

int main()
{
    //freopen("F:\\test.txt","r",stdin);
	//freopen("F:\\output.txt","w",stdout);
	while(cin>>n&&n!=0)
	{
		for(int i=0;i<n;i++)
			cin>>a[i];
		int wholeWeight=huffmanTree();
		cout<<wholeWeight<<endl;
	}
    return 0;
}


posted on 2013-03-05 11:22  cjweffort  阅读(285)  评论(0编辑  收藏  举报