蓝桥杯练习 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。
输入格式
输入的第一行包含一个正整数n(n<=100)。
接下来是n个正整数,表示p0, p1, …, pn-1,每个数不超过1000。
输出格式
输出用这些数构造Huffman树的总费用。
样例输入
5
5 3 8 2 9
样例输出
59
思路分析:使用小堆顶来解决,每次取的是最小的数,然后将两最小数求和插入回堆中,在JAVA中常PriorityQueue实现堆。说一下JAVA中的PriorityQueue(优先队列)的使用。
PriorityQueue类:优先队列
1.构造方法:
public PriorityQueue():默认大小以及排序方法,包装类从小到大(JAVA已写好)顺序,不能存放自己写的类
public PriorityQueue(int initialCapacity):指定大小
public PriorityQueue(Comparator<? super E> comparator):指定比较方法,使用默认大小:11
public PriorityQueue(int initialCapacity,Comparator<? super E> comparator):指定大小和比较方法
2.插入元素:add,offer,向队中插入元素
3.删除元素:remove,poll,获取并删除队首元素
一种特殊的删除方法:remove(object o),删除队列中和o相等的元素。
4.获取队首元素:peek方法
那么这道题目就先构造小堆顶,每次取出堆顶元素处理,代码如下:
import java.util.*;
public class Main {
//比较器,构造小根堆
static Comparator<Integer> comparator = new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o1-o2;
}
};
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int sum = 0;
int n = sc.nextInt();
//堆的构造
PriorityQueue<Integer> pq = new PriorityQueue<>(comparator);
for(int i = 0;i<n;i++){
pq.offer(sc.nextInt());
}
while (pq.size()>1){
//每次获取两个最小元素
int a = pq.poll();
int b = pq.poll();
int temp = a+b;
//记录总和
sum += temp;
pq.offer(temp);
}
System.out.println(sum);
}
}
这道题目我之前的写的代码如下:
import java.util.Arrays;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int[] num = new int[n];
for(int i = 0;i<n;i++) {
num[i] = scanner.nextInt();
}
int sum = 0;
for(int i = n;i>1;i--) {
num = getResult(num);
sum += num[0];
}
System.out.println(sum);
}
private static int[] getResult(int[] num) {
// TODO Auto-generated method stub
Arrays.sort(num);
num[1] = num[0]+num[1];
int[] temp = new int[num.length-1];
for(int i = 1;i<num.length;i++) {
temp[i-1] = num[i];
}
return temp;
}
}
PS:堆是一种很重要的数据结构,建议不会的话可以看一下数据结构,理解这种数据结构的特点,以及他的用途:堆排序以及求解TopK的问题。真的很重要,面试也经常问到。