算法题-根据字典value的比例返回key - Python

题目描述：

给定一个字典，例如：

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))

欢迎指正哦。