题目:给定一个整数序列,存放在数组xs中,长度为n,请计算该序列的熵。
熵的计算方法:对于一个长度为n的序列xs,它包含m+1中不同的取值,s0, s1, …, sm,这些取值对应的出现概率分别是p0, p1, …,pm,则这个序列的熵为H(X) = -(p0log2(p0) + p1log2(p1) + … + pm*log2(pm)). 其中,某个取值出现的概率p的计算方法为:这个取值出现的次数 除以 长度n。
入口方法:
double entropy(int[] xs) {
}
参考答案(未获得测试案例,仅供参考):
package test;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
public class Main3 {
public static void main(String[] args){
int[] xs=new int[] {1,3,2,3,2,2};
Main3 m=new Main3();
System.out.println(m.entropy(xs));
}
double entropy(int[] xs) {
//数组中的value作为map的key,map的value代表重复次数
//(int)Math.ceil(log2(xs.length))只是优化,防止扩容带来的性能开销
Map<Integer, Integer> map=new HashMap<>((int)Math.ceil(log2(xs.length)));
for (int i = 0; i < xs.length; i++) {
Integer times = map.get(xs[i]);
if(times==null) {
map.put(xs[i], 1);
}else {
map.put(xs[i], times+1);
}
}
return calculate(map, xs.length);
}
public double calculate(Map<Integer, Integer> map,int total) {
double res=0;
double p=0;
Collection<Integer> values = map.values();
for (Integer times : values) {
p=(times+0.0)/total;
res-=p*log2(p);
}
return res;
}
//log2(N)=loge(N)/loge(2),loge(N)代表以e为底的N的对数,loge(2)代表以e为底的2的对数
public double log2(double N) {
return Math.log(N)/Math.log(2);//Math.log的底为e
}
}
本次输出:
1.4591479170272446