九度OJ-1172-哈夫曼树

题目描述:
哈夫曼树,第一行输入一个数n,表示叶结点的个数。需要用这些叶结点生成哈夫曼树,根据哈夫曼树的概念,这些结点有权值,即weight,题目需要输出所有结点的值与权值的乘积之和。
输入:
输入有多组数据。
每组第一行输入一个数n,接着输入n个叶节点(叶节点权值不超过100,2<=n<=1000)。
输出:
输出权值。
样例输入:
5
1 2 2 5 9
样例输出:
37

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
int cmp(const void * a,const void *b)
{
    return *(int *)b-*(int *)a;
}
/*
模仿word2vec里创建Huffman树的方法
*/
int main()
{
    freopen("input.txt","r",stdin);
    int n;
    int * array;
    int pos1,pos2,a;
    int min1_index,min2_index;
    while(scanf("%d",&n)!=EOF){
        //printf("%d \n",n );
        array =(int *) malloc((2*n-1)*sizeof(int));
        for (int i = 0; i < n; ++i)
            scanf("%d",array+i);
        qsort(array,n,sizeof(int),cmp);
        for (int i = n; i < 2*n-1; ++i)
            array[i]=1000;
        //for (int i = 0; i < 2*n-1; ++i)   printf("%d ",array[i]);
        pos1 = n-1;
        pos2 = n;
        for (a = 0; a < n-1; ++a)
        {
            if(pos1>=0){
                if (array[pos1]<array[pos2])
                {
                    min1_index=pos1;
                    pos1--;
                }else{
                    min1_index=pos2;
                    pos2++;
                }

            }else{
                min1_index=pos2;
                pos2++;

            }
            if(pos1>=0){
                if (array[pos1]<array[pos2])
                {
                    min2_index=pos1;
                    pos1--;
                }else{
                    min2_index=pos2;
                    pos2++;
                }

            }else{
                min2_index=pos2;
                pos2++;
            }
            array[n+a]=array[min1_index]+array[min2_index];         
        }
        int sum=0;
        for (int i = n; i < 2*n-1; ++i)
            sum+=array[i];
        printf("%d\n", sum);


    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值