codeup 1921 Problem C哈夫曼编码

题目描述

哈弗曼编码大家一定很熟悉吧（不熟悉也没关系，自己查去。。。）。现在给你一串字符以及它们所对应的权值，让你构造哈弗曼树，从而确定每个字符的哈弗曼编码。当然，这里有一些小规定：

1.规定哈弗曼树的左子树编码为0，右子树编码为1；

2.若两个字符权值相同，则ASCII码值小的字符为左孩子，大的为右孩子；

3.创建的新节点所代表的字符与它的做孩子的字符相同；

4.所有字符为ASCII码表上32-96之间的字符（即“ ”到“`”之间的字符）。

输入

输入包含多组数据（不超过100组）
每组数据第一行一个整数n，表示字符个数。接下来n行，每行有一个字符ch和一个整数weight，表示字符ch所对应的权值，中间用空格隔开。
输入数据保证每组测试数据的字符不会重复。

输出

对于每组测试数据，按照输入顺序输出相应的字符以及它们的哈弗曼编码结果，具体格式见样例。

样例输入

3
a 10
b 5
c 8
4
a 1
b 1
c 1
d 1

样例输出

a:0
b:10
c:11
a:00
b:01
c:10
d:11

AC代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define max 100

typedef struct
{
    int parent;
    int left;
    int right;
    int weight;
    char sym;
}HTree;

typedef struct Node
{
    int start;
    char s[max];
}HCode;

void select(HTree H[],int n,int *s1,int *s2)
{
    //找到最大和第二大的结点
    int i;
    int min1=-1,min2=-1;//用于记最小值下表
    int flag1=1,flag2=1;
    for(i=0;i<n;i++)
    {
        if(flag1==1&&H[i].parent==-1)
        {
            min1=i;
            flag1=0;
        }//记录第一个出现的根节点
        if(H[i].parent==-1&&H[i].weight<H[min1].weight)min1=i;//更新最小结点
    }//求得最小weight元素的下标
    for(i=0;i<n;i++)
    {
        if((flag2==1)&&(i!=min1)&&(H[i].parent==-1))//要求i!=min1
        {
            min2=i;
            flag2=0;
        }//记录第一个出现的根节点
        if((H[i].parent==-1)&&(i!=min1)&&(H[i].weight<H[min2].weight))min2=i;//更新最小结点
    }//求得次小weight元素下表
    *s1=min1;
    *s2=min2;
}

void InitializeHTree(HTree H[],int n)//对已建好的H[]进行初始化
{//n个叶子结点
    int m=2*n-1;
    int s1,s2;
    for(int i=0;i<n;i++)
    {//初始化n个叶子结点
        H[i].left=-1;//-1表示没有孩子
        H[i].right=-1;
        H[i].parent=-1;//表示没有父节点
        scanf("%c %d",&H[i].sym,&H[i].weight);//输入数据
        getchar();
        //printf("%c %d\n",H[i].sym,H[i].weight);
    }
    for(int i=n;i<m;i++)
    {
        H[i].left=-1;//-1表示没有孩子
        H[i].right=-1;
        H[i].parent=-1;//表示没有父节点
        //非叶子结点没有char内容，weight在之后进行初始化
    }
    //对其余n-1个结点进行初始化,共n-1次循环
    for(int i=n;i<m;i++)
    {
        select(H,i,&s1,&s2);//收集s1和s2的值
        //选好了s1,s2之后，开始初始化
        H[s1].parent=i;
        H[s2].parent=i;
        H[i].left=s1;
        H[i].right=s2;
        H[i].weight=H[s1].weight+H[s2].weight;//得到权重
    }
    //两个结点都好了
}

void CreateCode(HTree H[],HCode C[],int n)
{
    int k,p;//k为当前结点下标,p为其父节点下标
    for(int i=0;i<n;i++)
    {
        C[i].s[max-1]='\0';
        C[i].start=max-1;
        k=i;
        p=H[k].parent;
        while(p!=-1)//i结点不为根结点
        {
            if(H[p].left==k)C[i].s[--C[i].start]='0';
            else C[i].s[--C[i].start]='1';
            k=p;
            p=H[k].parent;
        }
    }
}

void ShowCode(HTree H[],HCode C[],int n)
{
    int j;
    for(int i=0;i<n;i++)
    {
        printf("%c:",H[i].sym);
        for(j=C[i].start;C[i].s[j]!='\0';j++)
        {
            printf("%c",C[i].s[j]);
        }
        printf("\n");
    }
}

int main()
{
    //freopen("input.txt","r",stdin);
    int N;
    while(scanf("%d",&N)!=EOF)
    {
        getchar();//使用完scanf时使用getchar()吸收空格
        HTree H[2*N-1];
        HCode C[N];
        InitializeHTree(H,N);
        CreateCode(H,C,N);
        ShowCode(H,C,N);
    }
}