九度oj:求哈夫曼树（小顶堆数据结构）

最新推荐文章于 2020-06-06 18:51:09 发布

mysterious_i

最新推荐文章于 2020-06-06 18:51:09 发布

阅读量332

点赞数

分类专栏： oj堆操作

本文链接：https://blog.youkuaiyun.com/mysterious_i/article/details/78217533

版权

oj堆操作专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用优先级队列（小顶堆）来构造霍夫曼树的方法，以此来求解最优带权路径长度的问题。通过不断合并权值最小的两个节点，最终形成一棵霍夫曼树，并计算出总的带权路径长度。

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

#include <queue>
#include <stdio.h>
using namespace std;
priority_queue<int,vector<int>,greater<int> > Q;//建立小顶堆
//默认的大顶堆priority_queue<int> Q;
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF) {
      while(Q.empty() == false) Q.pop(); //每次清一个元素，所以用while
      for(int i=1;i<=n;i++) { //输入n个叶子节点的权值
        int x;
        scanf("%d",&x);
        Q.push(x);  //将权值放入堆中
      }
      int ans =0 ; //带权路径初始化为0
      while(Q.size()>1) {  //当堆中元素大于1
         int a = Q.top(); //top为堆中最小的元素
         Q.pop();   //删除节点
         int b = Q.top();
         Q.pop();
         ans += a + b; //累加权值
         Q.push(a+b); //将左右节点的权值之和放入集合F中

      }
      printf("%d\n",ans);

    }
    return 0;
}