哈夫曼树的带权路径长度 - STL - priority_queue(单调队列)

最新推荐文章于 2021-10-12 23:35:28 发布

HyperDai

最新推荐文章于 2021-10-12 23:35:28 发布

阅读量2.1k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： C/C++ STL 数据结构文章标签： stl

本文链接：https://blog.youkuaiyun.com/deaidai/article/details/73561844

该博客探讨了如何使用STL中的priority_queue（默认大顶堆）来解决哈夫曼树带权路径长度问题。通过自定义运算符实现小顶堆，博主提供了三种解法：直接构建哈弗曼树并累加路径长度、维护排序数组以及利用优先队列。样例输入和输出展示了不同方法的应用，并提醒注意溢出问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

STL中提供的priority_queue默认是大顶堆，可以通过自己重载运算符来实现小顶堆的优先队列

题目描述

给定n个权值作为n个叶子结点，构造哈夫曼树, 求其带权路径长度。

输入

输入由多组数据组成。
每组数据分成两行。第一行仅一个整数n（2<=n<=100000)。第二行有n个空格分开的权值，值范围在[1,1000000000]之间。

输出

对于每组测试数据，输出一行，即其对应哈夫曼树的带权路径长度对1000000007取模。

样例输入

4
7 5 2 4
8
5 29 7 8 14 23 3 11

样例输出

35
271

提示

注意运算溢出

题解

方法一.先建哈夫曼树再累加求和

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

#define N 100005
#define M (2*N-1)

typedef struct HTNode{
  
  // 树中结点的结构
    long long weight=0;
    int parent=0,lchild=0,rchild=0;
}HTNode,*HuffmanT;
HTNode ht[M+1];
void Select(HTNode ht[], int k, int &s1, int &s2){
// ht[1...k]中选择parent为0，并且weight最小的两个结点，其序号由变量s1，s2指示