EOJ 1815 Huffman树

#include <iostream>#include <stdlib.h>#include <stdio.h>#define max 1005struct node//建立二叉树{    int data;    struct node*lchild;    struct node*rchild;};typedef struct node NODE;int a[max];int w[2*max-1];//内部节点外部节点数目之和int ans=0;NODE *create_huffman_tree(int *a,int *w,int n){    NODE* addr[2*max-1];//存储每个内部外部节点    int n1,n2,i,j;    int u,v,min1,min2;    for(i=0;i<n;i++)//产生n个叶子结点    {        addr[i]=(NODE*)malloc(sizeof(NODE));        addr[i]->data=a[i];        addr[i]->rchild=NULL;        addr[i]->lchild=NULL;        w[i]=-w[i];//权值取负，表示未合并    }    for(i=n;i<2*n-1;i++)//产生n-1个内部节点    {        n1=-1;//当前最小值的下标        min1=99999999;//当前最小值的初值        n2=-1;//当前次小值得下标        min2=99999999;//当前次小值的初值        for(j=0;j<i;j++)        {            v=w[j];            u=-v;//真实的权值为-w[j]            if(u>0)//如果u>0，表示还未合并            {                if(u<min1)//如果比最小值小，将最小值设置为u，次小值置为刚才的最小值                {                    min2=min1;                    n2=n1;                    min1=u;                    n1=j;                }                else if(u<min2)//如果u比次小值小，将min2置为u                {                    min2=u;                    n2=j;                }            }        }        addr[i]=(NODE*)malloc(sizeof(NODE));//产生一个内部节点        addr[i]->data=-1;//值为-1标示此节点为内部节点        addr[i]->lchild=addr[n1];//左右子树为次小值，最小值        addr[i]->rchild=addr[n2];        w[i]=w[n1]+w[n2];//此内部节点的权值为子权值之和        w[n1]=-w[n1];//将两个权值置为正，标示已合并        w[n2]=-w[n2];    }    w[2*n-2]=-w[2*n-2];//将根节点的权值置为正    return(addr[2*n-2]);}void solve(NODE *temp,int i){    if(temp->data==-1)//如果此节点为内部节点，递归调用    {        i++;        solve(temp->lchild,i);        solve(temp->rchild,i);    }    else{        ans+=i*w[temp->data];//如果为外部节点，权值相加    }}int main(){    memset(a,0,sizeof(a));//将a，w函数初始化    memset(w,0,sizeof(w));    int n,i;    scanf("%d",&n);//输入节点个数    for(i=0;i<n;i++)    {        scanf("%d",&w[i]);//输入每个节点的权值        a[i]=i;    }    NODE *temp = create_huffman_tree(a,w,n);    solve(temp,0);    printf("%d\n",ans);    return 0;    }

Huffman树中有几个小技巧要记住。第一，默认权值全为正值，开始时将所有权值置为负值，标示未合并，当使用该节点后，将权值置为正值，标示已合并。第二，生成内部节点时，此节点应该为权值为负，且权值的负数最小和次小的两个节点，生成节点后要把节点的值置为-1用以和外部节点区分。第三，求最小造价值的时候要判断该节点值是否为-1来判断是否递归调用，也就是说需要把所有的外部节点的权值*路径长度相加。