HDOJ-2527

Safe Or Unsafe

问题描述
Javac++ 一天在看计算机的书籍的时候，看到了一个有趣的东西！每一串字符都可以被编码成一些数字来储存信息，但是不同的编码方式得到的储存空间是不一样的！并且当储存空间大于一定的值的时候是不安全的！所以Javac++ 就想是否有一种方式是可以得到字符编码最小的空间值！显然这是可以的，因为书上有这一块内容--哈夫曼编码(Huffman Coding)；一个字母的权值等于该字母在字符串中出现的频率。所以Javac++ 想让你帮忙，给你安全数值和一串字符串，并让你判断这个字符串是否是安全的？
 

Input
输入有多组case，首先是一个数字n表示有n组数据，然后每一组数据是有一个数值m(integer)，和一串字符串没有空格只有包含小写字母组成！
 

Output
如果字符串的编码值小于等于给定的值则输出yes，否则输出no。
 

Sample Input
2
12
helloworld
66
ithinkyoucandoit

 

Sample Output
no
yes

 

题解：

   题目考察哈夫曼树以及哈夫曼编码，将一个字符串进行哈夫曼编码后压缩成的一个新的字符串，新的字符串的长度是否比那个安全数值小。

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

public class Main {
	public static void main(String[] args) {
		Scanner cin = new Scanner(System.in);
		int n = cin.nextInt();
		while (n-- > 0) {
			int safe = cin.nextInt();
			String str = cin.next();
			int arr[] = new int[30];
			for (int i = 0; i < str.length(); i++) {
				char ch = str.charAt(i);
				arr[ch - 'a']++;
			}
			PriorityQueue<Integer> priorityQueue = new PriorityQueue<>();
			for (int i = 0; i < arr.length; i++) {
				priorityQueue.add(arr[i]);
			}
			int sum = 0;
			while (priorityQueue.size() != 1) {
				int a = priorityQueue.poll();
				int b = priorityQueue.poll();
				int flag = a + b;
				sum += flag;
				priorityQueue.add(flag);
			}
			if (sum == 0) {
				sum = str.length();
			}
			if (sum <= safe) {
				System.out.println("yes");
			} else {
				System.out.println("no");
			}
		}
	}
}