面试机试之求解字符串熵值

今天做了这道题目，信息熵很早就学过了，所以题目也不陌生了，简单来说，信息熵的计算公式为：某一事件出现的概率乘以它的对数形式的结果的负数就是该时间的信息熵，把一个集合里面的所有的事件的信息熵都加起来就行了，就可以得到了总的信息熵了，明白了这一点就很容易做这道题目了，因为计算量很小，所以代码几乎就是直接写出来的没有做一点性能上的优化，不过结果AC了，应该是输入的数据量规模也是不大的：

#!/usr/bin/env  python
#coding:utf-8
from __future__ import division
'''
计算给定字符的二进制信息熵（如有小数，保留至小数位后7位）                               
输入
输入任意一串字符
样例输入
aaaabbcd
输出
计算出字符串的信息熵
样例输出
1.75
'''
import math

string=raw_input()
str_list=list(string)
n=len(str_list)
str_list_single=list(set(str_list))
num_list=[]
for i in str_list_single:
    num_list.append(str_list.count(i))
list_two=zip(str_list_single,num_list)
entropy=0
for j in range(len(list_two)):
    entropy+=-1*(float(list_two[j][1]/n))*math.log(float(list_two[j][1]/n),2)
if len(str(entropy).split('.')[-1])>=7:
    print '%.7f' %entropy
else:
    print entropy

本地运行结果为：

aaaabbcd
1.75

abcdefgh
3.0

wwwgggbbbhhjj
2.2954657

rghjiolpdfghjkllll
3.1916116

提交结果为：

说明：
所有测试数据均已输出正确结果！
^_^