算法笔记练习 9.8 哈夫曼树 问题 C: 哈夫曼树 - 超级详细的思路讲解

算法笔记 专栏收录该内容
143 篇文章 1 订阅

算法笔记练习 题解合集

本题链接

题目

题目描述
哈夫曼树,第一行输入一个数n,表示叶结点的个数。需要用这些叶结点生成哈夫曼树,根据哈夫曼树的概念,这些结点有权值,即weight,题目需要输出所有结点的值与权值的乘积之和。

输入
输入有多组数据。
每组第一行输入一个数n,接着输入n个叶节点(叶节点权值不超过100,2<=n<=1000)。

输出
输出权值。

样例输入

2
2 8 
3
5 11 30 

样例输出

10
62

思路

代码主体框架用的是这道题的:问题 B: 算法6-13:自顶向下的赫夫曼编码

讲解思路之前先明确两个词的定义:

  1. 结点的值:指的是输入数据的时候得到的叶结点的值;
  2. 结点的权值:指的是这个结点到根结点的距离,且根结点的权值为 0。

下文中这两个词不会混用,可以严格按照上面的定义理解。


相比问题 B 的代码需要做一些改动:

  1. 问题 B 的代码在遍历的时候为了方便,直接把结点中代表结点的值的成员w归零,用作一个指路的变量,而这个题的w一直到最后都是有用的,遍历的时候必须保留,所以加入新的成员path用来指路,w全程都代表结点的值;
  2. 本题要求获得所有叶结点的权值和其值乘积的和,加入新的成员depth代表结点的权值。
  3. 遍历的时候每当向左走或者向右走之前,为左孩子或者右孩子结点计算其depth,大小即为父结点的depth加 1;
  4. 统计答案的地方应该出现在原本生成huffmanCode的地方。

具体实现参考下方代码,从 50 行开始,huffmanTree数组已生成完毕,开始统计答案。

代码

#include <iostream>
#include <vector>
#include <algorithm>

struct Node {
	int w, parent, lchild, rchild, path, depth;
};

void findMinTwo(std::vector<Node>& huffman, int& min1, int& min2) {
	int i = 1, w1, w2;
	while (huffman[i++].parent != 0)
		continue;
	min1 = i - 1;
	while (huffman[i++].parent != 0)
		continue;
	min2 = i - 1;
	for ( ; i < huffman.size(); ++i) { 
		if (huffman[i].parent == 0 && huffman[i].w < std::max(huffman[min1].w, huffman[min2].w)) { 
			if (huffman[min1].w != huffman[min2].w)
				(huffman[min1].w > huffman[min2].w ? min1 : min2) = i;
			else
				(min1 > min2 ? min1 : min2) = i; 
		}
	}
	if (min1 > min2)
		std::swap(min1, min2);
} 

int main() {
	int n, input, min1, min2;
	std::vector<Node> huffmanTree;
	while (scanf("%d", &n) != EOF) {
		if (n == 1) {
			scanf("%d", &input);
			printf("1\n");
			continue;
		} 
		huffmanTree.clear();
		huffmanTree.resize(n + 1);
		huffmanTree[0] = {0, 0, 0, 0, 0, 0};
		for (int i = 1; i <= n; ++i) {
			scanf("%d", &input);
			huffmanTree[i] = {input, 0, 0, 0, 0, 0};
		}
		for (int i = n + 1; i <= 2 * n - 1; ++i) {
			findMinTwo(huffmanTree, min1, min2);
			huffmanTree[min1].parent = huffmanTree[min2].parent = i;
			huffmanTree.push_back( {huffmanTree[min1].w + huffmanTree[min2].w, 0, min1, min2, 0, 0} );
		}
		int c = 2 * n - 1, sum = 0;
		while (c) {
			if (huffmanTree[c].path == 0) {
				huffmanTree[c].path = 1;
				if (huffmanTree[c].lchild != 0) {
					huffmanTree[huffmanTree[c].lchild].depth = huffmanTree[c].depth + 1;
					c = huffmanTree[c].lchild;
				} else if (huffmanTree[c].rchild == 0) {
					sum += huffmanTree[c].depth * huffmanTree[c].w;
				} 
			} else if (huffmanTree[c].path == 1) {
				huffmanTree[c].path = 2;
				if (huffmanTree[c].rchild != 0) {
					huffmanTree[huffmanTree[c].rchild].depth = huffmanTree[c].depth + 1;
					c = huffmanTree[c].rchild;
				} 
			} else {
				huffmanTree[c].path = 0;
				c = huffmanTree[c].parent;
			} 
		}
		printf("%d\n", sum); 
	} 
	return 0;
} 
  • 0
    点赞
  • 0
    评论
  • 1
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值