题意:给出字符出现的次数,问替换成哈夫曼编码后的文本长度。
实际上观察发现就等于树的所有节点的和。用nlogn超时。用O(n),用两个队列,一个放原始数组,一个放新生成的节点。
#include <iostream> #include <cstdio> #include <cmath> #include <algorithm> #include <vector> #include <iomanip> #include <cstring> #include <map> #include <queue> #include <set> #include <cassert> #include <stack> #define mkp make_pair using namespace std; const double EPS=1e-8; typedef long long lon; const lon SZ=50,INF=0x7FFFFFFF; lon arr[SZ]; int main() { std::ios::sync_with_stdio(0); //freopen("d:\\1.txt","r",stdin); lon casenum; //cin>>casenum; //for(lon time=1;time<=casenum;++time) { lon n; cin>>n; queue<lon> q1,q2; for(lon i=0;i<n;++i) { lon tmp; cin>>tmp; q1.push(tmp); } lon res=0; for(;;) { lon tmp=0; if(q1.size()==0&&q2.size()==1)break; for(int i=0;i<2;++i) { if(q1.empty()) { tmp+=q2.front(); q2.pop(); } else if(q2.empty()) { tmp+=q1.front(); q1.pop(); } else { if(q1.front()<=q2.front()) { tmp+=q1.front(); q1.pop(); } else { tmp+=q2.front(); q2.pop(); } } } q2.push(tmp); res+=tmp; } cout<<res<<endl; } return 0; }