1.问题分析
哈夫曼树又是最优二叉树,它的度为2(几个孩子),因此可以推算出它的总元素个数为2n-1。用数组法时可以由此确定出初始化元素个数。
解决方法为:先排序,再选择两个最小的数相加,结果放在队尾,然后删除用掉的这两个数。循环此操作,直至剩下最后一个数即可。
2.代码实现
2.1数组队列法
import java.util.Arrays;
import java.util.Scanner;
public class Main {
//用队列构造哈夫曼树
public static void main(String[] args) {
//1.初始化数组队列,输入数据
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();//个数
//哈夫曼树的度为2,加上根节点,所以总数为2n+1
if(n>0&&n<=100){
int[] queueArr=new int[2*n-1];
for(int i=0;i<n;i++){
queueArr[i]=sc.nextInt();
}
sc.close();
//2.创建队头和队尾指针变量front和rear,因为是数组,所以rear的实际位置要减1
int sum=0;//(总费用)
int front=0;
int rear=n-1;
//3.队列循序计算
while (true){
//1.对队列中的数据进行排序(升序)
Arrays.sort(queueArr,front,rear+1);
//2.尾指针后移
rear++;
//3.判断尾指针是否到达末尾
if(rear==2*n-1){
break;
}
//4.将最小两数总费用入队
queueArr[rear]=queueArr[front]+queueArr[front+1];
sum=sum+queueArr[rear];
//5.将最小的两个数出队
front=front+2;
}
if (n==1){
System.out.println(queueArr[0]);
}else {
System.out.println(sum);
}
}
}
}
2.2集合法
import java.util.*;
public class Demo22 {
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
if(n<=100){
List<Integer> list=new ArrayList<Integer>();
for(int i=0;i<n;i++){
list.add(sc.nextInt());
}
int sum=0;
while(list.size()>1){
Collections.sort(list);
int min1= (int) list.get(0);
int min2= (int) list.get(1);
list.remove(0);
list.remove(0);
list.add(min1+min2);
sum=sum+min1+min2;
}
System.out.println(sum);
}
}
}
2.3运行测试
可见两者运行时间和空间相差不多,论代码量和逻辑思路,还是集合比较方便。
ps:Arrays.sort(queueArr,front,rear+1);参数一为数组名,参数二为起始位置(不含本身),参数为结束位置(含)。