哈夫曼树 - 九度教程第30题

最新推荐文章于 2024-08-28 21:17:33 发布

原创

最新推荐文章于 2024-08-28 21:17:33 发布 · 845 阅读

3 ·

CC 4.0 BY-SA版权

本博客探讨了哈夫曼树的概念及其在九度教程第30题中的应用。题目要求根据输入的叶节点数量和权值，计算所有结点权值与数值乘积的总和。通过使用优先队列和堆数据结构，可以实现高效的哈夫曼树构建，达到O(nlogn)的时间复杂度。哈夫曼树在哈夫曼编码和数的两两合并等问题中有着广泛应用。

哈夫曼树 - 九度教程第30题

题目

时间限制：1 秒内存限制：32 兆特殊判题：否
题目描述：
哈夫曼树，第一行输入一个数n，表示叶结点的个数。需要用这些叶结点生成哈夫曼树，根据哈夫曼树的概念，这些结点有权值，即weight，题目需要输出所有结点的值与权值的乘积之和。
输入：
输入有多组数据。
每组第一行输入一个数n，接着输入n个叶节点（叶节点权值不超过100，2<=n<=1000）。
输出：
输出权值。
样例输入：
5
1 2 2 5 9
样例输出：
37
来源：
2010年北京邮电大学计算机研究生机试真题

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

using namespace std;

//建立一个小顶堆
priority_queue<int, vector<int>, greater<int> > Q;

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF){
        while(Q.empty()==false)Q.pop();//清空堆中元素
        for(int i=1;i<=n;i++){
            //输入n个叶子节点权值
            int x;
            scanf("%d",&x);
            Q.push(x);//将权值放入堆中
        }

        int ans=0;//保存答案
        while(Q.size()>1){//当堆中元素大于1个
            int a=Q.top();
            Q.pop();
            int b=Q.top();
            Q.pop();