在数据结构的课上学习了哈夫曼树,书上有实现哈夫曼树的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;
}