藍橋杯-基礎練習 Huffman樹（python與C++版）

試題 基礎練習 Huffuman樹

資源限制
時間限制：1.0s 內存限制：512.0MB
問題描述
　　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

題比較簡單，先上python代碼：

1、python

# coding=utf-8

all = 0
huf = []
n = int(input())
huf =input().split()
huf = list(map(int,huf))
while len(huf)>1:
	minx = (min(huf))
	huf.remove(minx)
	all += minx
	if len(huf)==0:
		break
	miny = min(huf)
	all+=miny
	huf.remove(miny)
	huf.append(minx+miny)
print(all)

2、C++

#include<iostream>
#define MIN 100000
using namespace std;

int popmin(int a[],int &len){
	int m = MIN,pos=0;
	for(int i=0;i<len;i++){
		if(a[i]<m){
			m = a[i];
			pos = i;
		}
	}
	for(int i=pos;i<len;i++){
		a[i]=a[i+1];
	}
	len--;
	return m;
}


int main()
{	
	int n,huf[105],sum=0;
	cin>>n;
	for(int i=0;i<n;i++)
		cin>>huf[i];
	int m1,m2;
	while(n>1){
		m1 = popmin(huf,n);
		sum+=m1;
		if(n==0) break;
		m2 = popmin(huf,n);
		sum+=m2;
		huf[n]=m1+m2;
		n++;
	}
	cout<<sum<<endl;
	return 0;
}