EOJ 1815 Huffman树

#include <iostream>
#include <stdlib.h>
#include <stdio.h>
#define max 1005
struct 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来判断是否递归调用，也就是说需要把所有的外部节点的权值*路径长度相加。