382. 链表随机节点

原理参考:蓄水池抽样——《编程珠玑》读书笔记
思路:以1/m的概率选择第m个对象
思路

class Solution:
    def __init__(self, head: ListNode):
        p = head
        self.vals = []
        while p:
            self.vals.append(p.val)
            p = p.next
        self.n = len(self.vals)

    def getRandom(self) -> int:
        import random
        index = random.randrange(0,self.n)
        return self.vals[index]

在这里插入图片描述
进阶:
如果链表十分大且长度未知,如何解决这个问题?你能否使用常数级空间复杂度实现?

class Solution:
    def __init__(self, head: ListNode):
        self.head = head
        
    def getRandom(self) -> int:
        import random
        ans,cur,count = self.head.val,self.head.next,1
        while cur:
            count+=1
            if random.randrange(0,count)==0: #这个成立的概率是1/count
                ans = cur.val
            cur = cur.next
        return ans

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值