水塘抽样，相等概率随机取数（不定长数据流）

水塘抽样，相等概率随机取数（不定长数据流）

问题描述：设 $nums$ 中有 $k$ 个值为 $target$ 的元素，保证这 $k$ 个值成为返回值的概率均为 $\frac{1}{k}$

操作：遍历 $nums$，当我们第 $i$ 次遇到值为 $target$ 的元素时，随机选取 $[0,i)$ 内的一个整数，如果其等于 $0$ (范围内任何值都可)，那么就将返回值置为该元素下标，否则不变。

证明：
$$\begin{array}{rl}& P(第i次遇到值为target的元素的下标成为最终返回值)\\ & =P(第i次随机=0)\times P(第i+1次随机!=0)\times \dots \times P(第k次随机!=0)\\ & =\frac{1}{i}\times (1-\frac{1}{i+1})\times \dots (1-\frac{1}{k})\\ & =\frac{1}{i}\times \frac{i}{i+1}\times \dots \times \frac{k-1}{k}\\ & =\frac{1}{k}\end{array}$$

class Solution:
    def __init__(self, head: Optional[ListNode]):
        self.root = head

    def getRandom(self) -> int:
        node, ans, i = self.root, None, 0
        while node:
            if not randint(0, i):
                ans = node.val
            node, i = node.next, i + 1
        return ans