Java实现 LeetCode 710 黑名单中的随机数（黑白名单）

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 没有参数，输入参数是一个列表，即使参数为空，也会输入一个 [] 空列表。

class Solution {

	 int n=0, cnt=0;
    HashSet<Integer> h=null;
    boolean [] set=null;
    int [] candidates=null;
    public Solution(int N, int[] blacklist) {

      n=N;

      int L=blacklist.length, i=0;
      if( N>20000)
      {
        h=new HashSet<Integer>();

        while( i<L )
        {
          h.add(blacklist[i]);
          i++;
        }
        return;
      }

      set=new boolean [N];
      while( i<L )
      {
    	set[blacklist[i]]=true;
        i++;
      }
      i=0;

      cnt=N-L;
      candidates=new int [cnt];
      int j=0;
      while( i<N )
      {
    	if( set[i]==false )
    	{
    	  candidates[j]=i;
    	  j++;
    	}
        i++;
      }
    }

    public int pick() {

      if( n>20000 )
      {
    	int oup=(int)(Math.random()*n);

        while( h.contains(oup) )
          oup=(int)(Math.random()*n);

        return oup;
      }     

      return candidates[(int)(Math.random()*cnt)];
    }
}

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