oj --哈夫曼树-求带权路径长度

我是研究僧i 提交于 2020-01-31 07:04:45

题目所描述的有点问题,其实是叶节点的路径与权值乘积之和,即带权路径长度!

用到了c++中小顶堆的创建方法,优先队列。

定义:priority_queue<Type, Container, Functional>

Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector),Functional 就是比较的方式,当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆

还利用到:若定义每个结点都有值,为其子女的值之和,那么除了根结点外所有结点之和恰好等于带权路径长度,按层级累加的规律不难理解叶子节点的值也是“乘了好多倍”

在这里插入图片描述

#include<iostream>
#include<stdio.h>

#include<queue>

using namespace std;
priority_queue<int,vector<int>,greater<int> > Q;
int main(){
	int N;
	while(scanf("%d",&N)!=EOF){
		while(!Q.empty()) Q.pop();
		for(int i = 0; i < N; i++){
			int x;
			scanf("%d",&x);
			Q.push(x);
		}
		int ans = 0;
		while(!Q.empty()&&Q.size()>1){
			int a = Q.top();
			Q.pop();
			int b = Q.top();
			Q.pop();
			ans += a + b;
			Q.push(a+b);
		}
		printf("%d\n",ans);
	}
}

没有必要真的构造一个哈夫曼树,用小顶堆每次输出即可,构造一个虚拟的哈夫曼树。

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!