[HDOJ5350]MZL's munhaff function

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5350

智商题…求最小哈夫曼树的编码。取数据中最小的两个数，相加并加到总和中，将相加后的和作为一个节点。

用到优先队列。

比如case2，最小哈夫曼树是这样的。

            104
          /     
         50     54
        /     / 
      25  25  28  26
      /
     1 24

 将所有非叶节点的值相加即可。

#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstring>
#include <climits>
#include <complex>
#include <fstream>
#include <cassert>
#include <cstdio>
#include <bitset>
#include <vector>
#include <deque>
#include <queue>
#include <stack>
#include <ctime>
#include <set>
#include <map>
#include <cmath>

using namespace std;

priority_queue<int, vector<int>, greater<int> > pq;

int main() {
    // freopen("in", "r", stdin);
    int T, n;
    int tmp;
    scanf("%d", &T);
    while(T--) {
        while(!pq.empty()) {
            pq.pop();
        }
        scanf("%d", &n);
        for(int i = 0; i < n; i++) {
            scanf("%d", &tmp);
            pq.push(tmp);
        }
        long long ans = 0;
        int a, b;
        while(n-- > 1) {            
            a = pq.top();
            pq.pop();
            b = pq.top();
            pq.pop();
            tmp = a + b;
            pq.push(tmp);
            ans += tmp;
        }
        printf("%I64d
", ans);
    }
}

#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstring>
#include <climits>
#include <complex>
#include <fstream>
#include <cassert>
#include <cstdio>
#include <bitset>
#include <vector>
#include <deque>
#include <queue>
#include <stack>
#include <ctime>
#include <set>
#include <map>
#include <cmath>

using namespace std;

priority_queue<int, vector<int>, greater<int> > pq;

int main() {
    // freopen("in", "r", stdin);
    int T, n;
    int tmp;
    scanf("%d", &T);
    while(T--) {
        while(!pq.empty()) {
            pq.pop();
        }
        scanf("%d", &n);
        for(int i = 0; i < n; i++) {
            scanf("%d", &tmp);
            pq.push(tmp);
        }
        long long ans = 0;
        int a, b;
        while(n-- > 1) {            
            a = pq.top();
            pq.pop();
            b = pq.top();
            pq.pop();
            tmp = a + b;
            pq.push(tmp);
            ans += tmp;
        }
        printf("%I64d
", ans);
    }
}