Leetcode习题集-黑名单中的随机数

题目：

给定一个包含 [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]

---

我的答案：

private List<Integer> list ;
Random random;

public Solution(int N, int[] blacklist) {
    random = new Random();
    Set<Integer> set = new HashSet<>();
    for (int i = 0; i < N; i++) {
        set.add(i);
    }
    for (int a : blacklist) {
        set.remove(a);
    }
    list = new ArrayList<>(set);
}
public int pick() {
    return list.get(random.nextInt(list.size()));
}

结果超出内存限制

---

标准答案：

分析：
白名单中数的个数为 N - len(B)，那么可以直接在 [0, N - len(B)) 中随机生成整数。我们把所有小于 N - len(B) 且在黑名单中数一一映射到大于等于 N - len(B) 且出现在白名单中的数。这样一来，如果随机生成的整数出现在黑名单中，我们就返回它唯一对应的那个出现在白名单中的数即可。
例如当 N = 6，B = [0, 2, 3] 时，我们在 [0, 3) 中随机生成整数，并将 2 映射到 4，3 映射到 5，这样随机生成的整数就是 [0, 1, 4, 5] 中的一个。

算法：
我们将黑名单分成两部分，第一部分 X 的数都小于 N - len(B)，需要进行映射；第二部分 Y 的数都大于等于 N - len(B)，这些数不需要进行映射，因为并不会随机到它们。
我们先用 Y 构造出 W，表示大于等于 N - len(B) 且在白名单中的数，X 和 W 的长度一定是相等的。随后遍历 X 和 W，构造一个映射表（HashMap）M，将 X 和 W 中的数构造一一映射。
在 [0, N - len(B)) 中随机生成整数 a 时，如果 a 出现在 M 中，则将它的映射返回，否则直接返回 a。

代码：

class Solution {
    Map<Integer, Integer> m;
    Random r;
    int wlen;
    public Solution(int n, int[] b) {
        m = new HashMap<>();
        r = new Random();
        wlen = n - b.length;
        Set<Integer> w = new HashSet<>();
        for (int i = wlen; i < n; i++) w.add(i);
        for (int x : b) w.remove(x);
        Iterator<Integer> wi = w.iterator();
        for (int x : b)
            if (x < wlen)
                m.put(x, wi.next());
    }
    public int pick() {
        int k = r.nextInt(wlen);
        return m.getOrDefault(k, k);
    }
}

---