PTA 7-8（树） 修理牧场（25 分） 最优二叉树

最优二叉树问题

把最小的两端拿出来连接后放入队列中，重复的上述操作，，

用优先队列可以容易完成

（注：博客作为交流使用，切勿抄袭应付作业）

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <cmath>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <ctype.h>
#include <vector>
#include <algorithm>
#include <sstream>
#define PI acos(-1.0)
#define in freopen("in.txt", "r", stdin)
#define out freopen("out.txt", "w", stdout)
using namespace std;
typedef long long ll;
const int maxn = 1000 + 7, INF = 0x3f3f3f3f, mod = 1e9 + 7;
int n;
priority_queue<int, vector<int>, greater<int> > qu;

void init() {
scanf("%d", &n);
int x;
for(int i = 0; i < n; ++i) {
scanf("%d", &x);
qu.push(x);
}
}

void solve() {
int ans = 0;
while(!qu.empty()) {
int t1 = qu.top(); qu.pop();
if(qu.empty()) { break; }
int t2 = qu.top(); qu.pop();
t1 += t2;
ans += t1;
if(qu.empty()) break;
qu.push(t1);
}
cout << ans << endl;
}

int main() {
init();
solve();
return 0;
}
/*
8
4 5 1 2 1 3 1 1
*/