LeetCode刷题——随机数索引#398#Medium

随机数索引的思路探讨与源码
    随机数索引的题目如下图，该题属于数学类和哈希表类型的题目，主要考察对于数学方法的使用和哈希表方法的理解。本文的题目作者想到2种方法，分别是哈希表方法和模拟方法，其中哈希表方法使用Java进行编写，而模拟方法使用Python进行编写，当然这可能不是最优的解法，还希望各位大佬给出更快的算法。

    本人认为该题目可以使用模拟方法的思路进行解决，首先初始化一个字典和随机数类，然后开始循环遍历数组，将数组元素和对应的初始化列表放入字典，如果存在相同的KEY值就不放入字典，再把当前下标的值存入当前列表中。在选取函数里面，是直接从字典里选出一个传入值对应的列表，然后通过随机数来找一个值做KEY值后找到列表对应的元素并返回结果。那么按照这个思路我们的Java代码如下：

#喷火龙与水箭龟
class Solution {
    Map<Integer, List<Integer>> posMap;
    Random random;

    public Solution(int[] nums) {
        posMap = new HashMap<Integer, List<Integer>>();
        random = new Random();
        for (int ir = 0; ir < nums.length; ir++) {
            posMap.putIfAbsent(nums[ir], new ArrayList<Integer>());
            posMap.get(nums[ir]).add(ir);
        }
    }

    public int pick(int target) {
        List<Integer> indMap = posMap.get(target);
        return indMap.get(random.nextInt(indMap.size()));
    }
}

    显然，我们的哈希表方法的效果比较一般，还可以使用模拟的方法解决。首先初始化一个字典并进行遍历，然后通过随机数来寻找字典里对应的元素的值并返回。所以按照这个思路就可以解决，下面是Python代码：

#喷火龙与水箭龟
class Solution:

    def __init__(self, nums: List[int]):
        self.map = defaultdict(list)
        for i, num in enumerate(nums):
            self.map[num].append(i)

    def pick(self, target: int) -> int:
        return random.choice(self.map[target])

    从结果来说Java版本的哈希表方法的效率一般，而Python版本的模拟方法的速度还不错，但应该是有更多的方法可以进一步提速的，希望朋友们能够多多指教，非常感谢。