(哈夫曼树10.3.1)POJ 3253 Fence Repair(计算带权路径长度最小的哈夫曼树)

/*
 * POJ_3253.cpp
 *
 *  Created on: 2013年11月7日
 *      Author: Administrator
 */

#include <iostream>
#include <cstdio>
#include <algorithm>


using namespace std;

const int maxn = 20010;

long n,len;//目标模板数、堆长
long long p[maxn];//堆


void heap_insert(long long k){//将k插入小根堆,并维护小根堆的性质
	long t = ++len;//将k插入堆尾
	p[t] = k;
	while(t > 1){//自下而上将k插入到合适的位置上
		if(p[t/2] > p[t]){//如果p[t]的值<其父的值
			swap(p[t/2],p[t]);//交换位置
			t /= 2;
		}else{
			break;
		}
	}
}


void heap_pop(){//取小根堆的堆首元素,并维护小根堆的性质
	long t = 1;
	p[t] = p[len--];
	while(t*2 <= len){//从堆首开始,自上而下调整
		long k = t*2;
		if( k < len && p[k] > p[k+1]){//计算左右孩子中值较小的节点序号k
			++k;
		}

		if(p[t] > p[k]){//将k的值与其父比...
			swap(p[t],p[k]);
			t = k;
		}else{
			break;
		}
	}
}


int main(){
	while(scanf("%lld",&n)!=EOF){
		long i;
		for(i = 1 ; i <= n ; ++i){
			scanf("%lld",&p[i]);
		}

		for(i = 1 ; i <= n ; ++i){
			heap_insert(p[i]);
		}

		long long ans = 0;
		while(len > 1){
			long long a,b;
			a = p[1];
			heap_pop();
			b = p[1];
			heap_pop();

			ans += (a+b);
			heap_insert((a+b));
		}

		printf("%lld\n",ans);
	}

	return 0;
}