710. Random Pick with Blacklist

最新推荐文章于 2025-05-14 07:15:00 发布

weixin_34178244

最新推荐文章于 2025-05-14 07:15:00 发布

阅读量99

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/bright-mark/p/9603938.html

本文深入探讨了在存在黑名单的情况下，从[0,N)区间内生成随机数的算法实现。作者尝试通过蓄水池算法去除黑名单元素，但在实现过程中遇到了超时等问题，暗示可能需要更高效的算法来解决此问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：

生成[0,N)的数组，然后减去blacklist的名单，使用蓄水池算法，类似于382题的解法，但是最终超时了，也不知道有没有其他错误。

等待寻找更好的算法

class Solution {
public:
    Solution(int N, vector<int> blacklist) {
        
        for(int i=0;i<N;i++)mp[i]=i;
        
        for(int i=0;i<blacklist.size();i++){
            auto it=mp.find(blacklist[i]);
            if(it!=mp.end()) mp.erase(it);
        }
    }
    
    int pick() {
        int i=1;
        int choice=1;
        //for(auto it:mp){//这么写程序报错？问题得再找找
        for(map<int,int>::iterator it=mp.begin();it!=mp.end();it++){
            int j=rand()%i;
            if(j==0) choice=it->second;
            i++;
        }
        return choice;
    }
private:
    map<int,int>mp;
};