Huffman树——合并果子

原题链接

解析

这道题是一个典型的Huffman树的题目
对于任何一种构造都可以转化成一棵树,这道题相当于是求跟结点的权制最小
因此这道题可以用Huffman树来进行解决,权值越小的点离树根越远越好,权值越大的点离树根越近越好
因此可以用小根堆来实现这个过程

代码

#include <bits/stdc++.h>
using namespace std;
int n;
priority_queue<int,vector<int>,greater<int> >vis;
int main()
{
    
    
    scanf("%d",&n);
    while(n--)
    {
    
    
        int x;
        scanf("%d",&x);
        vis.push(x);
    }
    int res=0;
    while(vis.size()>1)
    {
    
    
        int a=vis.top();
        vis.pop();
        int b=vis.top();
        vis.pop();
        res+=(a+b);
        vis.push(a+b);
    }
    cout<<res<<endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_46126537/article/details/113103241
今日推荐