码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj 4198 [Noi2015]荷马史诗——哈夫曼树

时间:2018-08-01 18:58:40      阅读:120      评论:0      收藏:0      [点我收藏+]

标签:const   string   题目   scan   amp   ios   operator   top   define   

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=4198

学习一下哈夫曼树。https://www.cnblogs.com/Zinn/p/9400381.html

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define ll long long
using namespace std;
int n,k,ans;
ll sum;
struct Node{
    ll w;int dep;
    Node(ll w=0,int d=0):w(w),dep(d) {}
    bool operator< (const Node &b)const
    {return w==b.w?dep>b.dep:w>b.w;}//dep!!!
};
priority_queue<Node> q;
int main()
{
    ll x;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)scanf("%lld",&x),q.push(Node(x,0));
    while(n%(k-1)!=1&&k!=2)n++,q.push(Node(0,0));
    while(q.size()>1)
    {
        n++;x=0;int d=0;
        for(int j=1;j<=k;j++)
        {
            Node k=q.top();q.pop();
            x+=k.w;sum+=k.w;d=max(d,k.dep);//max!!
        }
        q.push(Node(x,d+1));
        if(q.size()==1)ans=d+1;
    }
    printf("%lld\n%d\n",sum,ans);
    return 0;
}

 

bzoj 4198 [Noi2015]荷马史诗——哈夫曼树

标签:const   string   题目   scan   amp   ios   operator   top   define   

原文地址:https://www.cnblogs.com/Narh/p/9403248.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!