2016网易校招笔试 霍夫曼树

首先想到的是霍夫曼编码，但是又不全是。利用构建霍夫曼树的基本原理，但是在建树计算权值时不是最小的两个权值相加，而是结合u、v计算新的权值，并将其并入。

我的代码如下：

import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.Map;
import java.util.Scanner;

public class Main {
  public static void main(String[] args) {
    Scanner scanner = new Scanner(System.in);
    int u = scanner.nextInt();    //输入短音时间
    int v = scanner.nextInt();    //输入长音时间
    while (true) {
      String string = scanner.next();    //输入字符串
      //String string="abbbccccdddddddeeeeeeeefffffffffgggggggggggg";   
      //权值为（1，3，4，7，8，9，12），根据我的方法结果为108
      //统计每个字符频数（因为题目中只需要计算长度，不需要得到最后编码，因此只保留权值）
      LinkedList<Integer> values = getCount(string);    
      int total = count(u, v, values);    //根据权值计算最短时间
      System.out.println(total*u);
    }

  }

  //计算所需最短时间
  private static int count(int u, int v, LinkedList<Integer> values) {
    int total = 0;
    while (values.size() != 1) {    //结束条件：只有一个节点
      Collections.sort(values);   //对权值数字排序

      //最小的乘以v+第二小的乘以u，得到新节点的权值。(应该做归一化处理，已更改。感谢tx校招群中@上交-软开-通信-周童鞋)
      //删除前两个节点的权值，将其频数和加入权值数组（类似于霍夫曼树构建的过程）
      total = values.removeFirst() * v/u + values.removeFirst();    
      values.add(total);   
    }
    return total;
  }

  //统计字符出现次数（权值）
  public static LinkedList<Integer> getCount(String string) {
    char[] charArray = string.toCharArray();
    Map<Character, Integer> map = new HashMap<Character, Integer>();
    for (char c : charArray) {
      if (map.get(c) == null) map.put(c, 1);
      else  map.put(c, map.get(c) + 1);
    }
    return new LinkedList<Integer>(map.values());
  }
}