POJ 1521 Entropy（哈夫曼树）

最新推荐文章于 2022-07-21 11:41:33 发布

_binbin

最新推荐文章于 2022-07-21 11:41:33 发布

阅读量802

点赞数

分类专栏：数据结构文章标签：哈夫曼树并查集

本文链接：https://blog.csdn.net/binwin20/article/details/9104709

版权

数据结构专栏收录该内容

23 篇文章 0 订阅

订阅专栏

题意：球把一个字符串用哈夫曼编码表示所需要的二进制位数。及压缩比。

用并查集查是否在统一棵树上。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <cmath>
#include <stack>
#include <map>
#include <string>
#define LL long long
#define DB double
#define SF scanf
#define PF printf
#define N 1000009
using namespace std;
char ch[N];
int v[299];
int s[299];
int fa[299];
int finfa(int k)
{
    if(k==fa[k]) return k;
    return fa[k] = finfa(fa[k]);
}
void un(int a,int b)
{
    if(finfa(a)==finfa(b)) return ;
    fa[finfa(a)] = finfa(b);
}
struct nod{
    int a,b,x,y;
    void ini(){a=b=0x3f3f3f3f;x=y=-1;}
    void in(int i,int val)
    {
        if(a>val) a = val,x = i;
        if(b>a) swap(a,b),swap(x,y);
    }
} ee;
void solve()
{
    int c = 200;
    for(int i=0;i<299;i++) fa[i] = i;
    while(1)
    {
        ee.ini();
        for(int i=0;i<c;i++)
        if(v[i]&&s[i]==0)
        {
            ee.in(i,v[i]);
        }
        if(ee.x==-1) break;
        for(int i=0;i<c;i++)
        if(finfa(i)==finfa(ee.x)||finfa(i)==finfa(ee.y)) s[i]++;
        v[c++] = v[ee.x]+v[ee.y];
        un(ee.x,ee.y);
        un(ee.x,c-1);
    }
    if(c==200) s[(int)ch[0]]++;
    int len = strlen(ch),ans = 0;
    for(int i=0;i<len;i++)
    {
        ans += s[(int)ch[i]];
    }
    PF("%d %d %.1lf\n",len*8,ans,len*8.0/ans);
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif
    while(1)
    {
        SF("%s",ch);
        if(strcmp("END",ch)==0) break;
        int len = strlen(ch);
        memset(v,0,sizeof(v));
        memset(s,0,sizeof(s));
        for(int i=0;i<len;i++)
        {
            v[(int)ch[i]]++;
        }
        solve();
    }
    return 0;
}