Leetcode 710. 黑名单中的随机数 C++

Leetcode 710. 黑名单中的随机数

题目

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

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

提示:

1. 1 <= N <= 1000000000
2. 0 <= B.length < min(100000, N)
3. [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 没有参数，输入参数是一个列表，即使参数为空，也会输入一个 [] 空列表。

题解

白名单中个数为N-BlackList.size，那么可以直接在[0，N-BlackList.size）中生成随机数。我们把所有小于N-BlackList.size且在黑名单中数一一映射到大于等于N-BlackList.size且出现在白名单中的数。详细过程见代码

代码

class Solution {
public:
    int whiteSize;		//白名单元素个数
    unordered_map<int,int> m;		//黑名单中的数到白名单数的映射
    Solution(int N, vector<int>& blacklist) {
        whiteSize = N - blacklist.size();
        unordered_set<int> x;
        for(int i=whiteSize; i<N; i++)  x.insert(i);		//大于等于N-BlackList.size
        for(int i=0; i<blacklist.size(); i++)		//删除黑名单中的数，则得到大于等于N-BlackList.size且出现在白名单中的数
            x.erase(blacklist[i]);
        unordered_set<int>::iterator iter = x.begin();
        for(int i=0; i<blacklist.size(); i++){			//设置映射
            if(blacklist[i] < whiteSize)
                m[blacklist[i]] =  *iter++;
        }
    }
    
    int pick() {
        int choice = random()%whiteSize;
        return m.count(choice)==0 ? choice : m[choice];		//如果m中有则说明是黑名单中的数，我们需要使用映射
    }
};

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

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