leetcode刷题（105）——710. 黑名单中的随机数

最新推荐文章于 2023-01-03 14:56:44 发布

进击的代码家

最新推荐文章于 2023-01-03 14:56:44 发布

阅读量209

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/u012124438/article/details/113737568

版权

leetcode 专栏收录该内容

131 篇文章 1 订阅

订阅专栏

本文介绍一种高效的算法，用于从给定的[0, n)整数范围内生成不在黑名单B中的随机数，通过减少对`Math.random()`的调用次数，适用于大规模数据和黑名单场景。通过构建映射表，确保在黑名单影响下仍保持随机性。

摘要由CSDN通过智能技术生成

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

白名单中数的个数为 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> map;
    int wlen;
    Random r;
    public Solution(int N, int[] blacklist) {
        map = new HashMap();
        wlen = N - blacklist.length;
        r = new Random();
        HashSet<Integer> set = new HashSet();
        for(int i=wlen;i<N;i++){
            set.add(i);
        }
        for(int i=0;i<blacklist.length;i++){
            if(set.contains(blacklist[i])){
                set.remove(blacklist[i]);
            }
        }
        Iterator<Integer> it = set.iterator();
        for(int i=0;i<blacklist.length;i++){
            if(blacklist[i]<wlen){
                map.put(blacklist[i],it.next());
            }
        }
    }
    
    public int pick() {
        int random = r.nextInt(wlen);
        return map.getOrDefault(random,random);
    }
}

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