python leetcode 382. Linked List Random Node

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

猜你喜欢

转载自blog.csdn.net/Neekity/article/details/84574042
今日推荐