[LeetCode Python3] 382. Linked List Random Node + 水塘抽样 +大数据流，等概率抽样

SincX

于 2021-04-05 09:56:41 发布

阅读量152

点赞数

分类专栏： LeetCode每日一题

本文链接：https://blog.youkuaiyun.com/xvrixingkong/article/details/115438366

版权

LeetCode每日一题专栏收录该内容

172 篇文章

订阅专栏

这篇博客介绍了如何利用‘水塘抽样’算法在大数据流中实现等概率选取k个元素，以解决类似LinkedListRandomNode问题。通过维护一个计数器和当前节点值，循环遍历链表并根据随机数确定保留值，最终返回随机选取的值。此方法适用于需要在大数据集中随机采样的场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

382. Linked List Random Node

class Solution:
    def __init__(self, head: ListNode):
        self.head = head
    def getRandom(self) -> int:
        count = 0
        reserve = self.head.val
        cur = self.head
        while cur:
            count += 1
            rand = random.randint(1, count)
            if rand == count:
                reserve = cur.val
            cur = cur.next
        return reserve