哈夫曼树(竞赛用板子)

在数据结构的课上学习了哈夫曼树,书上有实现哈夫曼树的C语言代码,但是对于打竞赛来说并不实用,对于打竞赛来说哈夫曼树算是比较简单的数据结构了,这么一个简单的数据结构再去敲上上百行不太值当。正好想起来之前在打牛客小白月赛的时候碰见过一道哈夫曼树的题,当时队友说这是道典型的哈夫曼树的题目,很简单,当时我根本就不知道哈夫曼树是啥所以一点也不觉得简单,现在再回过头来看这道题,就是哈夫曼树的板子题呗,哈夫曼树也不难,所以这道题确实很简单。先上题目链接:
https://ac.nowcoder.com/acm/contest/11212/D
在数据结构的课上学习了哈夫曼树但是并没有学哈夫曼树的实现,因此还得自己琢磨,感觉要建树但是应该又有简单的法子,因此去找队友要板子,队友说找不到了,还说哈夫曼树的题直接用优先队列模拟就可以了,因此结合着网上的代码,下面这一份板子也就诞生了。

#include <iostream>
#include <string>
#include <vector>
#include <queue>
using namespace std;
int cnt[30],deep[30];
struct Node
{
    int cnt;
    char c;
    string s;
};
priority_queue <Node,vector <Node> > q;
bool operator < (Node x,Node y)
{
    return x.cnt>y.cnt;
}
int main()
{
    string s;
    cin >> s;
    for(int i=0;i<s.size();i++)
        cnt[s[i]-'a'+1]++;
    for(int i=1;i<=26;i++)
    {
        if(cnt[i])
        {
            char c='a'+i-1;
            q.push(Node{cnt[i],c,""});
        }
    }
    int ans=0;
    if(q.size()==1)
        ans+=q.top().cnt;
    else
    {
        while(q.size()>=2)
    	{
        	Node x=q.top();
        	q.pop();
        	Node y=q.top();
        	q.pop();
        	string t="";
        	if(x.c!=0&&y.c!=0)
        	{
            	t+=x.c;
            	t+=y.c;
        	}
        	if(x.c!=0&&y.c==0)
        	{
            	t+=x.c;
            	t+=y.s;
        	}
        	if(x.c==0&&y.c!=0)
        	{
            	t+=x.s;
            	t+=y.c;
        	}
        	if(x.c==0&&y.c==0)
        	{
            	t+=x.s;
            	t+=y.s;
        	}
        	for(int i=0;i<t.size();i++)
            	deep[t[i]-'a'+1]++;
        	q.push(Node{x.cnt+y.cnt,0,t});
    	}
    	for(int i=1;i<=26;i++)
        	ans+=cnt[i]*deep[i];
    }
    cout << ans << endl;
    return 0;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值