基礎練習 Huffuman樹

問題描述

　　Huffman樹在編碼中有着廣泛的應用。在這裏，我們只關心Huffman樹的構造過程。
　　給出一列數{pi}={p0, p1, …, pn-1}，用這列數構造Huffman樹的過程如下：
　　1. 找到{pi}中最小的兩個數，設爲pa和pb，將pa和pb從{pi}中刪除掉，然後將它們的和加入到{pi}中。這個過程的費用記爲pa +pb。
　　2. 重複步驟1，直到{pi}中只剩下一個數。
　　在上面的操作過程中，把所有的費用相加，就得到了構造Huffman樹的總費用。
　　本題任務：對於給定的一個數列，現在請你求出用該數列構造Huffman樹的總費用。

　　例如，對於數列{pi}={5, 3, 8, 2, 9}，Huffman樹的構造過程如下：
　　1. 找到{5, 3, 8, 2, 9}中最小的兩個數，分別是2和3，從{pi}中刪除它們並將和5加入，得到{5, 8, 9, 5}，費用爲5。
　　2. 找到{5, 8, 9, 5}中最小的兩個數，分別是5和5，從{pi}中刪除它們並將和10加入，得到{8, 9, 10}，費用爲10。
　　3. 找到{8, 9, 10}中最小的兩個數，分別是8和9，從{pi}中刪除它們並將和17加入，得到{10, 17}，費用爲17。
　　4. 找到{10, 17}中最小的兩個數，分別是10和17，從{pi}中刪除它們並將和27加入，得到{27}，費用爲27。
　　5. 現在，數列中只剩下一個數27，構造過程結束，總費用爲5+10+17+27=59。

輸入格式

　　輸入的第一行包含一個正整數n（n<=100）。
　　接下來是n個正整數，表示p0, p1, …, pn-1，每個數不超過1000。

輸出格式

　　輸出用這些數構造Huffman樹的總費用。

樣例輸入

5
5 3 8 2 9

樣例輸出

59

解答代碼

#include<iostream>
#include<string>
#include<cstring>
#include<set>
#include<algorithm>
using namespace std;

#define N 1024

int main()
{
	int i,j,n,temp=0,data;
	multiset<long long> s;
	multiset<long long>::iterator pos;
	s.clear();
	cin>>n;
	for(i=0;i<n;i++)
	{
		cin>>data;
		s.insert(data);
	}
	long long result=0;
	if(n>1)
	{
		while(true)
		{
			if(s.size()<2)
				break;
			pos=s.begin();
			int a=(*pos);
			s.erase(pos);

			pos=s.begin();
			int b=(*pos);
			s.erase(pos);

			result+=(a+b);

			s.insert(a+b);		
		}
	}
	else
	{
		pos=s.begin();
		result+=(*pos);
	}
	cout<<result<<endl;
	return 0;
}