LeetCode刷题记录(二十三):随机数索引

79 阅读2分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第23天,点击查看活动详情

随机数索引

image.png

题目解析

题目素材解析

根据题目描述,我们可以看到本题目一共给出了两个传入参数。

但是不同于其他题目,本道题还给出了一个构造函数。

两个参数分别对应构造函数和实际方法。

  1. 可能含有重复元素的数组,元素为整数,传入构造函数。

  2. 一个整型数字,传入方法内。

我的解读

这道题与其他题对比起来,在日常中绝对是有更多复现的机会的。

先分析一下这道题的预期结果,从一个整数数组中,随机给出一个值的下标索引。

就比如{1,2,3,3,3}。

1的随机下标索引只能是0。

2的随机下标索引只能是1。

3的随机下标索引就可能会是2,3,4。

以此规律,保证每一次获取到的索引值是随机的。

Java中获取随机数的解决方法有很多,random就是一个很好用的类。

而如何取出每一个参数的下标组合呢?这个就需要先预处理一下整数数组了。

解决这两个问题,也就能解答出这道题了。

解题思路

解题思路也是围绕上面说的两点进行。

第一步,先预处理整数数组,转换成Map<Integer, List>集合。

第二步,根据传入的整数,通过Map集合获取到相应的索引下标集合。

第三部,随机取出List集合中的元素。

搞定收工。

代码

class Solution {

    Map<Integer, List<Integer>> data = new HashMap<Integer, List<Integer>>();
    Random random = new Random();

    public Solution(int[] nums) {
        for (int i = 0; i < nums.length; ++i) {
            List<Integer> list = data.get(nums[i]);
            if(list != null){
                list.add(i);
            }else{
                list = new ArrayList<Integer>();
                list.add(i);
                data.put(nums[i], list);
            }
        }
    }

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

/**
 * Your Solution object will be instantiated and called as such:
 * Solution obj = new Solution(nums);
 * int param_1 = obj.pick(target);
 */

执行结果

今天依旧不太景气,没有想到更好的方式,有待优化。

image.png