哈夫曼树

时间限制：1秒  空间限制：65536K  热度指数：3511

 算法知识视频讲解

题目描述

哈夫曼树，第一行输入一个数n，表示叶结点的个数。需要用这些叶结点生成哈夫曼树，根据哈夫曼树的概念，这些结点有权值，即weight，题目需要输出所有结点的值与权值的乘积之和。

输入描述:

输入有多组数据。
每组第一行输入一个数n，接着输入n个叶节点（叶节点权值不超过100，2<=n<=1000）。

输出描述:

输出权值。

示例1

输入

5  
1 2 2 5 9

输出

37

所有结点的值与权值的乘积之和=哈夫曼树的所有非叶结点的权值之和

package test;

import java.util.PriorityQueue;
import java.util.Scanner;

public class haffman {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner scanner=new Scanner(System.in);
		while (scanner.hasNext()) {
			int n=scanner.nextInt();
			PriorityQueue<Integer> queue=new PriorityQueue<Integer>();
			for (int i = 0; i < n; i++) {
				queue.add(scanner.nextInt());
			}
			int ans=0;
			while (queue.size()>1) {
				int first=queue.poll();
				int second=queue.poll();
				ans+=first+second;
				queue.add(first+second);				
			}
			System.out.println(ans);
		}
	}

}

使用了优先队列，对队列中的所有元素进行从小到大排序，根据哈夫曼树的构造规则，将哈夫曼树中的每次最小的两个结点相加成为新的结点。