382.[LeetCode]Linked List Random Node(蓄水池算法)

最新推荐文章于 2022-01-07 09:55:48 发布

薛昭君

最新推荐文章于 2022-01-07 09:55:48 发布

阅读量512

点赞数

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

本文链接：https://blog.csdn.net/qq_28057541/article/details/52335641

版权

LeetCode 专栏收录该内容

103 篇文章 0 订阅

订阅专栏

参考资料：非常的好的文章，一看就会

此题要使用的是经典的蓄水池算法，该方法的使用场景是 数据流长度未知，要求随机

public class Solution {

    private ListNode 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. 至少有一个node，默认链表不为空*/
    public Solution(ListNode head) {
        this.head = head;
    }

    /** Returns a random node's value. */
    // 这个方法的时间复杂度为 O(n)
    public int getRandom() {
        int count = 1;
        ListNode currentNode = head;
        int val = currentNode.val;
        Random r = new Random();
        while(currentNode != null){ // 这里是一个bug点，经常写成currentNode.next导致最后一个element没有进入操作
            // nextInt 方法生成一个 介于[0，n)的数，包含0，但是不包含n
            if(r.nextInt(count++) == 0){
                val = currentNode.val;
            }
            currentNode = currentNode.next;
        }
        return val;
    }
}