题意:给出字符出现的次数,问替换成哈夫曼编码后的文本长度。
实际上观察发现就等于树的所有节点的和。用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-;
typedef long long lon;
const lon SZ=,INF=0x7FFFFFFF;
lon arr[SZ]; int main()
{
std::ios::sync_with_stdio();
//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=;i<n;++i)
{
lon tmp;
cin>>tmp;
q1.push(tmp);
}
lon res=;
for(;;)
{
lon tmp=;
if(q1.size()==&&q2.size()==)break;
for(int i=;i<;++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 ;
}