-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathWeek_1_Day_05.py
45 lines (37 loc) · 1.34 KB
/
Week_1_Day_05.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
Q). Random Pick with Weight:
Given an array w of positive integers, where w[i] describes the weight of index i, write a function pickIndex
which randomly picks an index in proportion to its weight.
##Note:
1 <= w.length <= 10000
1 <= w[i] <= 10^5
pickIndex will be called at most 10000 times.
Example 1:
Input:
["Solution","pickIndex"]
[[[1]],[]]
Output: [null,0]
Example 2:
Input:
["Solution","pickIndex","pickIndex","pickIndex","pickIndex","pickIndex"]
[[[1,3]],[],[],[],[],[]]
Output: [null,0,1,1,1,0]
Explanation of Input Syntax:
The input is two lists: the subroutines called and their arguments. Solution's constructor has one argument,
the array w. pickIndex has no arguments. Arguments are always wrapped with a list, even if there aren't any.
Solution....................................................................................
import random
import bisect
class Solution:
def __init__(self, w: List[int]):
self.sumvalue=sum(w)
self.probablity=[-1]*len(w)
for i in range(len(w)):
w[i]+=0 if i==0 else w[i-1]
self.probablity[i]=w[i]/self.sumvalue
def pickIndex(self) -> int:
x=random.random()
index=bisect.bisect_left(self.probablity,x)
return index
# Your Solution object will be instantiated and called as such:
# obj = Solution(w)
# param_1 = obj.pickIndex()