POJ 3253 Fence Repair 哈夫曼树

#include <stdio.h>
#include <algorithm>
#include <set>
using namespace std;
int wood[20001],n;
typedef long long LL;
int main() {
while(scanf("%d",&n) != EOF) {
LL ans = 0;
multiset<int> S;
for(int i=0; i<n; i++) {
int x;
scanf("%d",&x);
S.insert(x);
}
// multiset默认就是按照从小到大排序的
while(S.size() > 1) {
multiset<int>::iterator it = S.begin();
int temp = *it; // 累加第一个元素
S.erase(it); // set中删除第一个元素
it = S.begin();
temp += *it; // 累加第二个元素
S.erase(it); // set中删除第二个元素
ans += temp;
S.insert(temp); // 将第一个第二个元素的和重新加入set
}
printf("%lld\n", ans);
}
return 0;
}