令我眼前一亮的代码(二)
开局一句话:
在(一)和(二)之间虽然隔了这么久,但是期间还是看到过很多有意思的小算法,小代码,很多都没有记录下来,或许觉得不够令我有记下来的冲动。
这次遇到的是我在写赫夫曼编码的时候,给你一段字符串,如果得到这个字符串中相同的字符出现的次数,即就是该字母在赫夫曼树中的权值,后面用来构建赫夫曼树,得到赫夫曼编码。
这并不是一个完整的程序,我只是觉得这个方法可能以后还会用到,于是将其记下,如果有更高效更好的方法,欢迎评论讨论。
代码如下:
/**
*
* @param bytes 接收字节数组
* @return 返回一个链表
*/
private static List<Node> getNodes(byte[] bytes){
//先创建一个ArrayList
ArrayList<Node> nodes=new ArrayList<Node>();
//遍历byte,统计每个byte出现的次数,使用map来得到[key,value]
Map<Byte,Integer>counts=new HashMap<>();
for (byte b:bytes){
Integer count=counts.get(b);
if (count==null){
//map还没有数据
counts.put(b,1);
}else{
counts.put(b,count+1);
}
}
//把每个键值对转成一个Node对象,并加入nodes集合
for (Map.Entry<Byte,Integer>entry:counts.entrySet()){
nodes.add(new Node(entry.getKey(),entry.getValue()));
}
return nodes;
}
说明:
因为在这个方法的程序中,创建赫夫曼树是将一个链表传进来构造成一颗树,所以这个方法最终得到是一个List,也可以根据不同的场景改变,重要的是其中的想法,先将给字符串转换为一个byte数组,传进这个方法,在创建一个hashmap,当然根据场景可以改变泛型,用一个for循环增强取出byte中的每个字符,开始在hashmap中取数据,当然一开始取不出来东西来或者说,因为哈希表中还没有存放着有数据,然后就将这个数据(字符)存放在哈希表中,这个时候这个字符的key值就是该字符的ASCII码,value是该字符在字符串中出现的次数,如果再次在字符串取到这个字符,因为相同的字符的ASCII值无疑都是相同的,再在哈希表中取这个key值时,即get(key)这个方法,就会取出上一次存进去的同一个字符,然后又将这个字符存进hashmap中,但是此时将它的value值加1了,嗯,主体差不多就是这样。