python leetcode 382. Linked List Random Node

最新推荐文章于 2021-04-05 09:56:41 发布

原创最新推荐文章于 2021-04-05 09:56:41 发布 · 636 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python

python 同时被 2 个专栏收录

263 篇文章

订阅专栏

leetcode

221 篇文章

订阅专栏

本文详细解析了蓄水池抽样算法的工作原理，通过实例演示如何在不知道总体长度的情况下，实现等概率随机抽样。算法适用于链表等数据结构，通过逐步迭代更新样本值，确保每个元素被选中的概率相等。

不知道长度蓄水池抽样原理？假设三个节点1,2,3,4
第一次循环 res=1
第二次循环 res有1/2的概率=2 另外1/2=1
第三次循环 res有1/3的概率=3 另外2/3的概率等于其他值根据第二循环可得 1/2*（2/3）=1/3的概率等于2
1/2*（2/3）=1/3的概率等于1
第四次循环 res有1/4的概率=4 另外3/4的概率等于其他值根据第三循环可得 1/3*（3/4）=1/4的概率等于3
1/3*（3/4）=1/4的概率等于2 1/3*（3/4）=1/4的概率等于1
一次类推

class Solution:
    import random 
    def __init__(self, head):
        """
        @param head The linked list's head.
        Note that the head is guaranteed to be not null, so it contains at least one node.
        :type head: ListNode
        """
        self.head=head

    def getRandom(self):
        """
        Returns a random node's value.
        :rtype: int
        """
        p=self.head
        if not p:
            return None
        i=0
        while p:
            if random.randint(0,i)==0:
                res=p.val
            i+=1
            p=p.next 
        return res