蓝桥杯基础训练 huffuman树构造问题

summerShouldBeNice

于 2021-03-24 11:27:20 发布

阅读量293

点赞数 2

文章标签： java 算法

本文链接：https://blog.csdn.net/qq_46736571/article/details/115166516

版权

huffuman树构造费用问题

问题描述：

Huffman树在编码中有着广泛的应用。在这里，我们只关心Huffman树的构造过程。
　　给出一列数{pi}={p0, p1, …, pn-1}，用这列数构造Huffman树的过程如下：
　　1. 找到{pi}中最小的两个数，设为pa和pb，将pa和pb从{pi}中删除掉，然后将它们的和加入到{pi}中。这个过程的费用记为pa + pb。
　　2. 重复步骤1，直到{pi}中只剩下一个数。
　　在上面的操作过程中，把所有的费用相加，就得到了构造Huffman树的总费用。
　　本题任务：对于给定的一个数列，现在请你求出用该数列构造Huffman树的总费用。

例如，对于数列{pi}={5, 3, 8, 2, 9}，Huffman树的构造过程如下：
　　1. 找到{5, 3, 8, 2, 9}中最小的两个数，分别是2和3，从{pi}中删除它们并将和5加入，得到{5, 8, 9, 5}，费用为5。
　　2. 找到{5, 8, 9, 5}中最小的两个数，分别是5和5，从{pi}中删除它们并将和10加入，得到{8, 9, 10}，费用为10。
　　3. 找到{8, 9, 10}中最小的两个数，分别是8和9，从{pi}中删除它们并将和17加入，得到{10, 17}，费用为17。
　　4. 找到{10, 17}中最小的两个数，分别是10和17，从{pi}中删除它们并将和27加入，得到{27}，费用为27。
　　5. 现在，数列中只剩下一个数27，构造过程结束，总费用为5+10+17+27=59。

先上代码，思路在下面

package Lanqiaobei;

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class Test020 {
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        //Integer.parseInt方法把接收的字符串转为整数
        int n = Integer.parseInt(br.readLine());
        //获取字符串
        String str = br.readLine();
        //将输入的数字以空格分开并加入数组
        String[] s = str.split(" ");
        //定义构造huffuman树用的费用
        int sum = 0;
        //定义int类型的数组方便后面用sort方法排序
        int[] a = new int[n];
        //因为费用会添加到数组中 所以定义一个变量保存每次构造huffuman树用的费用
        int num = 0;
        //遍历数组吧并且用parseInt方法将数据添加到数组里
        for (int i = 0; i < n; i++)
            a[i]=Integer.parseInt(s[i]);
        //使用sort方法给数组排序
        java.util.Arrays.sort(a);

        //如果获取的数组长度大于1
        if (n>1){
          //调用sort方法后 字符串是从小到大排序的 所以直接相加
         sum=a[0]+a[1];
         //费用赋值给num方便后面替换
         num = sum;
         //替换数据
         a[1] = num;
         //遍历
         for (int j =2;j<n;j++){
         	 //判断数组长度
             if (j<n){
                 java.util.Arrays.sort(a);
                 num=a[j-1]+a[j];
                 a[j]=num;
                 java.util.Arrays.sort(a);
                 sum+=num;
             }
         }
        }else {
            sum=a[0];
        }
        System.out.println(sum);
    }
}