【树】九度OJ 1172:哈夫曼树

    科技2022-08-17  103

    一、题目内容

    题目描述:

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

    输入:

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

    输出:

    输出权值。

    样例输入:

    5 1 2 2 5 9

    样例输出:

    37

    二、代码及注释

    #include<stdio.h> #include<queue> using namespace std; //思想,这里输出的是所有的中间节点权值和(非叶子节点),也叫带权路径长度,如1 2 2 5 9。输出(1+2)+(2+3)+(5+5)+(10+9)=37。 //先构成一个顶端为最小值的队列,然后每次取顶端两个值相加,再加入队列,直到队列中只剩一个值。 typedef long long ll; priority_queue<ll,vector<ll>,greater<ll>> Q; //这里为小顶堆 int main() { int n; scanf("%d",&n); while(!Q.empty()) Q.pop(); for(int i=0; i<n; i++) { ll x; scanf("%lld",&x); Q.push(x); } ll ans=0; while(Q.size()>1) { int a=Q.top(); Q.pop(); int b=Q.top(); Q.pop(); ans+=a+b; Q.push(a+b); } printf("%lld\n",ans); return 0; }
    Processed: 0.023, SQL: 9