Leetcode刷题710. 黑名单中的随机数

给定一个包含 [0，n ) 中独特的整数的黑名单 B，写一个函数从 [ 0，n ) 中返回一个不在 B 中的随机整数。

对它进行优化使其尽量少调用系统方法 Math.random() 。

提示:

1 <= N <= 1000000000
0 <= B.length < min(100000, N)
[0, N) 不包含 N，详细参见 interval notation 。
示例 1:

输入: 
["Solution","pick","pick","pick"]
[[1,[]],[],[],[]]
输出: [null,0,0,0]
示例 2:

输入: 
["Solution","pick","pick","pick"]
[[2,[]],[],[],[]]
输出: [null,1,1,1]
示例 3:

输入: 
["Solution","pick","pick","pick"]
[[3,[1]],[],[],[]]
Output: [null,0,0,2]
示例 4:

输入: 
["Solution","pick","pick","pick"]
[[4,[2]],[],[],[]]
输出: [null,1,3,1]
输入语法说明：

输入是两个列表：调用成员函数名和调用的参数。Solution的构造函数有两个参数，N 和黑名单 B。pick 没有参数，输入参数是一个列表，即使参数为空，也会输入一个 [] 空列表。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/random-pick-with-blacklist
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

感谢大神labuladong，传送门给我 O(1) 时间，我能查找/删除数组中的任意元素。

class Solution {

        int sz;
        Map<Integer, Integer> map;
        Random rand = new Random();

        //将黑名单中的数字交换到区间[sz,N)中，同时把[0,sz)中的黑名单数字映射到正常数字
        //时间复杂度O(B)，空间复杂度O(1)
        public Solution(int N, int[] blacklist) {
            map = new HashMap<>();
            //最终数组中的元素个数
            sz = N - blacklist.length;
            //将黑名单数字缓存到map，方便判断数字是否在黑名单内
            for (int b : blacklist) {
                map.put(b, -1);
            }
            //最后一个元素的索引
            int last = N - 1;
            //将黑名单中元素放到数组的最末尾
            for (int b : blacklist) {
                //如果b已经在区间[sz,N)可以直接忽略，不需要做映射
                if (b >= sz) {
                    continue;
                }
                //跳过黑名单内的数字，保证last不在黑名单
                while (map.containsKey(last)) {
                    last--;
                }
                map.put(b, last--);
            }
        }

        public int pick() {
            //随机返回区间[0,sz)一个数
            int num = rand.nextInt(sz);
            //如果这个数命中了黑名单，需要被映射到其他位置
            if (map.containsKey(num)) {
                return map.get(num);
            }
            //若没有命中黑名单，直接返回
            return num;
        }
    }