Pagini recente » Cod sursa (job #2243046) | Cod sursa (job #1255880) | Cod sursa (job #1565398) | Cod sursa (job #979845) | Cod sursa (job #791985)
Cod sursa(job #791985)
#include <fstream>
#define MAX 1048576
#define INF ((1LL << 62) - 1)
using namespace std;
struct nod
{
long long v;
long long f[2];
}nod[2 * MAX];
long long sol, b[MAX], minim;
int l[2], r[2], q[2][MAX], n, k, p, lgt[MAX];
void dfs(int nd, long long cod, long long l)
{
if(nod[nd].f[0])
{
dfs(nod[nd].f[0], cod * 2, l + 1);
dfs(nod[nd].f[1], cod * 2 + 1, l + 1);
return;
}
b[nd] = cod;
lgt[nd] = l;
}
void solve()
{
k = n;
l[0] = l[1] = 1;
r[0] = n;
while(l[0] + l[1] <= r[0] + r[1])
{
k++;
for(int i = 0; i < 2; i++)
{
p = 0; minim = INF;
for(int j = 0; j < 2; j++)
{
if(l[j] <= r[j] && nod[q[j][l[j]]].v < minim)
{
minim = nod[q[j][l[j]]].v;
p = j;
}
}
nod[k].f[i] = q[p][l[p]];
nod[k].v += nod[q[p][l[p]++]].v;
}
sol += nod[k].v;
q[1][++r[1]] = k;
}
dfs(k, 0, 0);
}
int main()
{
ifstream in("huffman.in");
in>>n;
for(int i = 1; i <= n; i++)
{
in>>nod[i].v;
q[0][i] = i;
}
in.close();
solve();
ofstream out("huffman.out");
out<<sol<<"\n";
for(int i = 1; i <= n; i++)
out<<lgt[i]<<" "<<b[i]<<"\n";
out.close();
return 0;
}