北大ACM3253 - Fence Repair（赫夫曼编码）

最新推荐文章于 2018-10-22 16:38:38 发布

原创最新推荐文章于 2018-10-22 16:38:38 发布 · 1.2k 阅读

0 ·

CC 4.0 BY-SA版权

北大ACM分析专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一种使用优先级队列实现赫夫曼编码的方法，并通过一个具体实例展示了如何构建赫夫曼树来最小化编码总长度。文章详细解释了赫夫曼编码原理及其在压缩算法中的应用。

1.1 算法分析

赫夫曼编码

由于这个问题，考察的是赫夫曼编码，所以优先级队列就直接使用STL的了。

1.2 代码

/*
 *
 * Introduction : ACM of pku
 * ID : 3253
 * alg : Huffman
 * Author : Gykimo
 * Date : 20121203
 * 
 */

#include <stdlib.h>
#include <stdio.h>
#include <queue>

using namespace std;

struct cmp
{
	bool operator ()(int x, int y){
		return x > y;
	}
};

int planks_num = 0;
int *planks = NULL;
long long min_amount = 0;

void readLine(){
	scanf("%d", &planks_num);
	planks = (int *)malloc(planks_num * sizeof(int));
	
	for(int i=0; i<planks_num; i++){
		scanf("%d", planks+i);
	}
}

void display(){
	printf("%lld\n", min_amount);
}

void huffman(){
	priority_queue<int, vector<int>, cmp> q(planks, planks+planks_num);

	int first, second, sum;
	while(q.size() > 1){
		first = q.top();
		q.pop();
		second = q.top();
		q.pop();
		sum = first + second;
		q.push(sum);
		min_amount += sum;
	}
}

int main(){
	readLine();
	huffman();
	display();
	return 0;
}