首先想到的是霍夫曼编码,但是又不全是。利用构建霍夫曼树的基本原理,但是在建树计算权值时不是最小的两个权值相加,而是结合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());
}
}