hdu1053~~hafman树~第一次写

题目连接：

http://acm.hdu.edu.cn/showproblem.php?pid=1053

题目意思：

输入一个字符串，每个字符是8位的，求一种编码，使得程度最小。

哈弗曼编码=权值*长度。

哈弗曼树框架：网上搜索到的，从下往上建树，不知道还有没有其他的算法。

每个点都是一个树，找最小权值的合并树，node包括value/parent/left/right

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 128
#define MAX 0xfffffff
using namespace std;
int num;
int f[N],mark[N];
int len[N];
char s[N];
struct node{
    int data,left,right,parent;
}haf[2*N+1];
void Haffman()
{
    int i,temp=num;
    for(i=0;i<num;i++){
        haf[i].data=f[i];//赋值
        haf[i].parent=-1;//表示空
        haf[i].left=-1;
        haf[i].right=-1;
    }
    while(true)
    {//num++往上冒，来建树
        int p1=-1,p2=-1;
        int min1=MAX,min2=MAX;
        for(i=0;i<num;i++){//等于-1，是为了？？？？把已经选了的点去掉，-1表示空节点
            if(haf[i].parent==-1 && haf[i].data<min1){p1=i;min1=haf[i].data;}
        }
        for(i=0;i<num;i++){
            if(haf[i].parent==-1&&haf[i].data<min2&&i!=p1){
                p2=i;min2=haf[i].data;
            }
        }
        if(p1==-1||p2==-1)break;
        haf[num].data=min1+min2;
        haf[num].left=p1;
        haf[num].right=p2;
        haf[num].parent=-1;
        haf[p1].parent=num;
        haf[p2].parent=num;
        num++;
    }
    for(i=0;i<temp;i++){//并非一条树
        len[i]=0;
        int t=i;
        while(haf[t].parent!=-1){
            len[i]++;
            t=haf[t].parent;
        }
    }
}
int main()
{
    int i,j,sum;
    while(scanf("%s",s),strcmp(s,"END")!=0)
    {
        num=sum=0;
        memset(f,0,sizeof(f));
        memset(mark,0,sizeof(mark));
        for(i=0;i<strlen(s);i++){mark[(int)s[i]]++;}
        for(i=0;i<N;i++){
            if(mark[i]!=0)f[num++]=mark[i];}
        if(num==1){
            printf("%d %d 8.0\n",8*strlen(s),f[0]);
            continue;
        }
        Haffman();
        for(i=0;i<num;i++)
            sum+=len[i]*f[i];
        printf("%d %d %.1lf\n",8*strlen(s),sum,8*(double)strlen(s)/(double)sum);
    }
    return 0;
}

题目代码：

很有技巧处理的代码：通过累加省去了频度*长度，直接用多次的累加把长度的相乘去掉了。

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
int a[30];
char s[1001];
int cmp(int a,int b)
{
    return a>b;
}
int main()
{
    while(gets(s))
    {
        if(strcmp(s,"END")==0)
        break;
        int m,n,i,j,k;
        m=strlen(s);
        memset(a,0,sizeof(a));
        for(i=0;i<m;i++)
        if(s[i]=='_')
        a[0]++;
        else
        a[s[i]-'A'+1]++;
        sort(a,a+27,cmp);
        for(i=0;;i++)
        if(a[i]==0)
        break;
        n=i;
        int t=0;
        for(i=n-1;i>0;i--)
        {
            a[i-1]=a[i]+a[i-1];
            t+=a[i-1];
            sort(a,a+i,cmp);
        }
        if(n==1)
        t=a[0];
        printf("%d %d %.1f\n",8*m,t,8.0*m/t);
    }
    return 0;
}