题目描述:
给定一个字典,例如:
symbolAndWeight = {'a': 2, 'b': 3, 'c': 5}
现在要求设计一个函数,每次调用这个函数返回一个key值,而且返回这个key值的概率与其对应的value成正比,其实就是key值对应的value越大,返回的概率就越大。
问题分析:
很显然这是一个几何概率的题目,把所有的value值,列出来摆成一条直线,并且确定对应的key值的区间,对这个整个区间进行随机生成一个整数,落在那个区间就返回那个key。
Python实现:
# 其实是一年前的面试题目 # @Time :2018/6/19 # @Author :LiuYinxing import random def randomS(symbolAndWeight): target = random.randint(0, sum(symbolAndWeight.values())) # 确定随机数 sum_ = 0 for k, v in symbolAndWeight.items(): sum_ += v if sum_ >= target: # 判断落在了那个区间 return k if __name__ == '__main__': symbolAndWeight = {'a': 2, 'b': 3, 'c': 5} print(randomS(symbolAndWeight))欢迎指正哦。