当前位置: 代码迷 >> 综合 >> 哈夫曼树 优先队列STL的应用
  详细解决方案

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

热度:67   发布时间:2024-01-20 20:20:35.0

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

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

这题添加虚拟节点,使得最后的合并方便,也就是使得每次合并保证树中都大于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;
}



  相关解决方案