题目

思路

很容易想到第一种做法，用Huffman算法，从森林中找出两个值最小的节点，合并再加入森林，在这个过程中不断记录。

但是每一次需要sort一遍，将最小的两个值节点置于头两个节点，最坏情况下复杂度是O(n^3)的量级，结果是TLE。

第二种方法是利用STL中的priority_queue模板进行解题，很方便，这里给出AC代码。但是优先队列手写实现还是需要实践，将在下一篇进行实现。

代码

Huffman：

//
//  main.cpp
//  Huffman Tree2
//
//  Created by wasdns on 16/12/24.
//  Copyright © 2016年 wasdns. All rights reserved.
//#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;struct Node
{int num;Node *l, *r;
};bool cmp(Node *n1, Node *n2)
{if (n1 -> num != n2 -> num) return n1 -> num < n2 -> num;return false;
}Node *node[100005];/*Ininode：初始化节点*/
void Ininode(int n)
{int i;for (i = 1; i <= n; i++){node[i] = new Node;cin >> node[i] -> num;node[i] -> l = NULL;node[i] -> r = NULL;}
}/*Huffman函数*/
int Huffman(int n)
{int cnt = 0;sort(node+1, node+n+1, cmp);while (1){if (node[2] -> num == 100005)                        //只剩下一棵树{break;}cnt += node[1] -> num;cnt += node[2] -> num;node[1] -> num = node[1] -> num + node[2] -> num;    //1、2合并为1node[2] -> num = 100005;                             //删除2sort(node+1, node+n+1, cmp);                         //将值最小的两棵树前移}return cnt;
}int main()
{int n;cin >> n;Ininode(n);int ans = Huffman(n);cout << ans << endl;return 0;
}

STL_Priority_Queue：

//
//  main.cpp
//  PriorityQueue STL
//
//  Created by wasdns on 16/12/24.
//  Copyright © 2016年 wasdns. All rights reserved.
//#include <iostream>
#include <functional>
#include <queue>
using namespace std;struct cmp
{bool operator() (int x,int y) {return x > y;}
};int storage[100005];int main()
{int n;cin >> n;//priority_queue<int, vector<int>, greater<int>> q;priority_queue<int, vector<int>, cmp> q;for (int i = 1; i <= n; i++){cin >> storage[i];q.push(storage[i]);}int cnt = 0;while (q.size() > 1){int x = q.top();q.pop();int y = q.top();q.pop();cnt += x+y;q.push(x+y);}cout << cnt << endl;return 0;
}

2016/12/24