【校招笔试】字符编码（哈夫曼树）

时间限制：1秒 空间限制：32768K 热度指数：629

本题知识点： 贪心

题目描述：

请设计一个算法，给一个字符串进行二进制编码，使得编码后字符串的长度最短。

输入描述

每组数据一行，为待编码的字符串。保证字符串长度小于等于1000。

输出描述

一行输出最短的编码后长度。

输入例子

MT-TECH-TEAM

输出例子

33

---

解题思路（以输入例子为例）

1. 统计各个字符出现的个数，如下：
   M(2), T(3), E(2), C(1), H(1), A(1), -(2)
2. 画出哈夫曼树，如下：
   
3. 发现编码后字符串的长度等于：各个非叶子结点值之和（2+3+5+4+7+12）
4. 利用这个，用HashMap存字符出现的个数，PriorityQueue优先队列选出两个最小的结点值

---

代码

import java.util.*;//比较器(没有用)class Comparators {    public static Comparator getComparator() {        return new Comparator() {            @Override            public int compare(Object o1, Object o2) {                if (o1 instanceof Integer) {                    return compare( (Integer) o1, (Integer) o2);                } else return 1;            }            public int compare(Integer s1, Integer s2) {                return s2-s1;            }        };    }}public class Main {    /*        测试代码     */    public static void main(String []args){        Scanner sc = new Scanner(System.in);        while(sc.hasNext()){            String str = sc.nextLine();            HashMap<Character, Integer> map = new HashMap<Character, Integer>();            for(int i = 0; i < str.length(); i++) {                Character c = str.charAt(i);                if (map.containsKey(c)) {                    map.put(c, map.get(c)+1);                } else {                    map.put(c, 1);                }            }            PriorityQueue<Integer> queue = new PriorityQueue<Integer>();            for(Character i : map.keySet()) {                queue.add(map.get(i));            }            int ans = 0;            while(!queue.isEmpty()) {                int num1 = queue.poll();                int num2 = queue.poll();                ans += (num1+num2);                if(!queue.isEmpty())queue.add(num1+num2);            }            System.out.println(ans);        }    }}