Class Solution
java.lang.Object
g0501_0600.s0528_random_pick_with_weight.Solution
528 - Random Pick with Weight.<p>Medium</p>
<p>You are given a <strong>0-indexed</strong> array of positive integers <code>w</code> where <code>w[i]</code> describes the <strong>weight</strong> of the <code>i<sup>th</sup></code> index.</p>
<p>You need to implement the function <code>pickIndex()</code>, which <strong>randomly</strong> picks an index in the range <code>[0, w.length - 1]</code> ( <strong>inclusive</strong> ) and returns it. The <strong>probability</strong> of picking an index <code>i</code> is <code>w[i] / sum(w)</code>.</p>
<ul>
<li>For example, if <code>w = [1, 3]</code>, the probability of picking index <code>0</code> is <code>1 / (1 + 3) = 0.25</code> (i.e., <code>25%</code>), and the probability of picking index <code>1</code> is <code>3 / (1 + 3) = 0.75</code> (i.e., <code>75%</code>).</li>
</ul>
<p><strong>Example 1:</strong></p>
<p><strong>Input</strong> [“Solution”,“pickIndex”] [<a href="1">1</a>,[]]</p>
<p><strong>Output:</strong> [null,0]</p>
<p><strong>Explanation:</strong></p>
<pre><code> Solution solution = new Solution([1]);
solution.pickIndex(); // return 0. The only option is to return 0 since there is only one element in w.
</code></pre>
<p><strong>Example 2:</strong></p>
<p><strong>Input</strong></p>
<pre><code> ["Solution","pickIndex","pickIndex","pickIndex","pickIndex","pickIndex"]
[[[1,3]],[],[],[],[],[]]
</code></pre>
<p><strong>Output:</strong> [null,1,1,1,1,0]</p>
<p><strong>Explanation:</strong></p>
<pre><code> Solution solution = new Solution([1, 3]);
solution.pickIndex(); // return 1. It is returning the second element (index = 1) that has a probability of 3/4.
solution.pickIndex(); // return 1
solution.pickIndex(); // return 1
solution.pickIndex(); // return 1
solution.pickIndex(); // return 0. It is returning the first element (index = 0) that has a probability of 1/4.
Since this is a randomization problem, multiple answers are allowed.
All of the following outputs can be considered correct:
[null,1,1,1,1,0]
[null,1,1,1,1,1]
[null,1,1,1,0,0]
[null,1,1,1,0,1]
[null,1,0,1,0,0]
......
and so on.
</code></pre>
<p><strong>Constraints:</strong></p>
<ul>
<li><code>1 <= w.length <= 10<sup>4</sup></code></li>
<li><code>1 <= w[i] <= 10<sup>5</sup></code></li>
<li><code>pickIndex</code> will be called at most <code>10<sup>4</sup></code> times.</li>
</ul>
-
Constructor Summary
Constructors -
Method Summary
-
Constructor Details
-
Solution
public Solution(int[] w)
-
-
Method Details
-
pickIndex
public int pickIndex()
-