为有牺牲多壮志,敢教日月换新天。

[Swift]LeetCode710. 黑名单中的随机数 | Random Pick with Blacklist

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号:山青咏芝(shanqingyongzhi)
➤博客园地址:山青咏芝(https://www.cnblogs.com/strengthen/
➤GitHub地址:https://github.com/strengthen/LeetCode
➤原文地址: https://www.cnblogs.com/strengthen/p/10506644.html 
➤如果链接不是山青咏芝的博客园地址,则可能是爬取作者的文章。
➤原文已修改更新!强烈建议点击原文地址阅读!支持作者!支持原创!
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

热烈欢迎,请直接点击!!!

进入博主App Store主页,下载使用各个作品!!!

注:博主将坚持每月上线一个新app!!!

Given a blacklist B containing unique integers from [0, N), write a function to return a uniform random integer from [0, N) which is NOT in B.

Optimize it such that it minimizes the call to system’s Math.random().

Note:

  1. 1 <= N <= 1000000000
  2. 0 <= B.length < min(100000, N)
  3. [0, N) does NOT include N. See interval notation.

Example 1:

Input: 
["Solution","pick","pick","pick"]
[[1,[]],[],[],[]]
Output: [null,0,0,0]

Example 2:

Input: 
["Solution","pick","pick","pick"]
[[2,[]],[],[],[]]
Output: [null,1,1,1]

Example 3:

Input: 
["Solution","pick","pick","pick"]
[[3,[1]],[],[],[]]
Output: [null,0,0,2]

Example 4:

Input: 
["Solution","pick","pick","pick"]
[[4,[2]],[],[],[]]
Output: [null,1,3,1]

Explanation of Input Syntax:

The input is two lists: the subroutines called and their arguments. Solution's constructor has two arguments, N and the blacklist Bpick has no arguments. Arguments are always wrapped with a list, even if there aren't any.


给定一个包含 [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 和黑名单 Bpick 没有参数,输入参数是一个列表,即使参数为空,也会输入一个 [] 空列表。


Runtime: 984 ms
Memory Usage: 23.4 MB
 1 class Solution {
 2     var M:Int = 0
 3     var map:[Int:Int] = [Int:Int]()
 4     init(_ N: Int, _ blacklist: [Int]) {
 5         var N = N
 6         for b in blacklist
 7         {
 8             map[b] = -1
 9         }
10         M = N - map.count
11         for b in blacklist
12         {
13             if b < M
14             {
15                 while (map[N - 1] != nil)
16                 {
17                     N -= 1
18                 }
19                 map[b] = N - 1
20                 N -= 1
21             }
22         }        
23     }
24     
25     func pick() -> Int {
26         var p:Int = Int.random(in:0..<M)
27         return map[p] == nil ? p : map[p]!      
28     }
29 }
30 
31 /**
32  * Your Solution object will be instantiated and called as such:
33  * let obj = Solution(N, blacklist)
34  * let ret_1: Int = obj.pick()
35  */ 

1628ms

 1 class Solution {
 2 
 3     init(_ N: Int, _ blacklist: [Int]) {
 4         M = N - blacklist.count
 5         for n in blacklist {
 6             mapping[n] = 0
 7         }
 8         
 9         var i = M
10         for n in blacklist {
11             if n < M {
12                 while mapping[i] != nil {
13                     i += 1
14                 }
15                 mapping[n] = i
16                 i += 1
17             }
18         }
19     }
20     
21     func pick() -> Int {
22         let r = Int.random(in: 0..<M)
23         if let n = mapping[r] {
24             return n
25         }
26         return r
27     }
28     
29     var M : Int
30     var mapping = [Int: Int]()
31 }
32 
33 /**
34  * Your Solution object will be instantiated and called as such:
35  * let obj = Solution(N, blacklist)
36  * let ret_1: Int = obj.pick()
37  */

 

posted @ 2019-03-10 19:38  为敢技术  阅读(261)  评论(0编辑  收藏  举报