哈夫曼树 优先队列STL的应用

最近每次做题都要看很多资料。看来是真的老了。

这题给定字符串，以及待编码的字符数，采用哈夫曼编码，编码总长度最短。求此值。

这题添加虚拟节点，使得最后的合并方便，也就是使得每次合并保证树中都大于T个。T为提供的编码字符数。

相当于每一个节点要么是空的要么是有T个孩子的。

设待编码字符为N，添加的虚节点个数为β。

那么对（N+β）进行一次合并-> (N+β)-T+1 第二次合并 （N+β）-2*（T-1）

......

最后一次(N+β)-r*(T-1)=1

也就是(N+β)%(T-1)==1%(T-1); 好像出现了同余

...

下面代码基本上是段学妹的.. 我待会儿看看STL的一些个东西...

PS:据说priority_queue内部是用红黑树实现的,跪了....

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<map>
using namespace std;

bool cmp( int a,int b ){
return a<b;
}

int main()
{
freopen( "K.in","r",stdin );
freopen( "ans.out","w",stdout );
char str[111111];
int n;
map<char,int> mp;
map<char,int>::iterator it;

priority_queue< int,vector<int>,greater<int> >que;
while( gets(str) )
{
scanf( "%d",&n );
getchar();
mp.clear();
for( int i=0;i<strlen(str);i++ )
mp[str[i]]++;

while( !que.empty() ) que.pop();

for( it=mp.begin();it!=mp.end();it++ )
que.push((*it).second);

while( que.size()%(n-1)!=1%(n-1))
que.push(0);

int ans=0;
if( mp.size()<=n )
ans = strlen(str);
else
{
while( que.size()!=1 )
{
int su=0;
for( int i=0;i<n;i++ )
{
su+=que.top();
que.pop();
}
ans+=su;
que.push(su);
}
}
printf( "%d\n",ans );
}
return 0;
}