leetcode398

考查数学概率

题目

方法一：

class Solution {

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

        public Solution(int[] nums) {
            for (int i = 0; i < nums.length; i++) {
                List<Integer> record = records.getOrDefault(nums[i], new ArrayList<>());
                record.add(i);
                records.put(nums[i], record);
            }
            random = new Random();
        }

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

方法二：

class Solution {

        private Random random;
        private int[] nums;

        public Solution(int[] nums) {
            this.nums = nums;
            random = new Random();
        }

        public int pick(int target) {
            int ans = 0;
            int cnt = 0;
            for (int i = 0; i < nums.length; i++) {
                if (nums[i] == target) {
                    ++cnt;
                    if (random.nextInt(cnt) == 0) {
                        ans = i;
                    }
                }
            }
            return ans;
        }
    }