【哈夫曼树】牛客哈夫曼树

最新推荐文章于 2024-06-14 08:58:42 发布

努力努力再努力0.0

最新推荐文章于 2024-06-14 08:58:42 发布

阅读量316

点赞数

分类专栏：牛客网

本文链接：https://blog.csdn.net/qq_36927714/article/details/86671544

版权

牛客网专栏收录该内容

16 篇文章 0 订阅

订阅专栏

题目描述

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

输入描述:

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

输出描述:

输出权值。

示例1

输入

5  
1 2 2 5 9

输出

#include <stdio.h>
#include <cstdio>
#include <string>
#include <string.h>
#include <math.h>
#include <algorithm>
#include <iostream>
#include <queue>
#include <vector>
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.size()!=1)
        {
            int a=q.top();
            q.pop();
            int b=q.top();
            q.pop();
            int tmp=a+b;
            ans=ans+tmp;
            q.push(tmp);
        }
        printf("%d\n",ans);
    }
}