Pagini recente » Cod sursa (job #1668422) | Cod sursa (job #634107) | Cod sursa (job #1100171) | Cod sursa (job #1191412) | Cod sursa (job #902609)
Cod sursa(job #902609)
#include <cstdio>
#include <queue>
#include <set>
#include <vector>
using namespace std;
struct node {
int index;
node *left;
node *right;
long long val;
node(int _index, long long _val, node *_left, node *_right):
index(_index), val(_val), left(_left), right(_right) {}
};
struct node_comp {
bool operator()(const node *a, const node *b) {
return a->val > b->val;
}
};
const int MAXN = (int)(1e6 + 2);
int len[MAXN];
int code[MAXN];
void dfs(node *root, int d, int val)
{
if (root->left == NULL && root->right == NULL) {
len[root->index] = d;
code[root->index] = val;
return;
}
dfs(root->left, d + 1, val << 1);
dfs(root->right, d + 1, (val << 1) + 1);
}
int main()
{
freopen("huffman.in", "r", stdin);
freopen("huffman.out", "w", stdout);
int n;
scanf("%d", &n);
priority_queue<node *, vector<node *>, node_comp> tree;
for (int i = 0; i < n; ++i) {
long long v;
scanf("%lld", &v);
tree.push(new node(i, v, NULL, NULL));
}
node *root;
long long total_value = 0;
while (true) {
node *low = tree.top();
tree.pop();
if (tree.empty()) {
root = low;
break;
}
node *next = tree.top();
tree.pop();
tree.push(new node(-1, low->val + next->val, low, next));
total_value += low->val;
total_value += next->val;
}
printf("%lld\n", total_value);
dfs(root, 0, 0);
for (int i = 0; i < n; ++i)
printf("%d %d\n", len[i], code[i]);
return 0;
}