python leetcode 382. Linked List Random Node

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

Neekity

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

阅读量564

点赞数

分类专栏： leetcode python 文章标签： leetcode python

本文链接：https://blog.csdn.net/Neekity/article/details/84574042

版权

python 同时被 2 个专栏收录

263 篇文章 0 订阅

订阅专栏

leetcode

221 篇文章 2 订阅

订阅专栏

不知道长度蓄水池抽样原理？假设三个节点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