398 随机数索引（哈希表）

1. 问题描述：

给定一个可能含有重复元素的整数数组，要求随机输出给定的数字的索引。 您可以假设给定的数字一定存在于数组中。
注意：
数组大小可能非常大。 使用太多额外空间的解决方案将不会通过测试。

示例:

int[] nums = new int[] {1,2,3,3,3};
Solution solution = new Solution(nums);
// pick(3) 应该返回索引 2,3 或者 4。每个索引的返回概率应该相等。
solution.pick(3);
// pick(1) 应该返回 0。因为只有nums[0]等于1。
solution.pick(1);
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/random-pick-index

2. 思路分析：

分析题目可以知道我们可以使用哈希表解决，因为使用的是python语言所以可以使用python中的字典来存储数字出现的下标，键为nums中的数字，值为数字出现的所有下标，我们可以使用collections.defaultdict(list)声明一个值为List的字典，这样值就为List类型就可以添加多个下标，在初始化的时候遍历nums，将nums中出现数字对应的下标添加到字典中即可，在调用pick函数的时候获取当前键为target的下标列表，使用random中的randint函数随机返回一个下标即可。

3. 代码如下：

from typing import List
import collections
import random


class Solution:
    # 使用哈希表存储即可
    def __init__(self, nums: List[int]):
        dic = collections.defaultdict(list)
        for i, x in enumerate(nums):
            dic[x].append(i)
        self.dic = dic

    def pick(self, target: int) -> int:
        dic = self.dic
        # 当前键为target的下标列表
        i = dic[target]
        # 随机返回键为target的某个下标
        return dic[target][random.randint(0, len(i) - 1)]