2022.01.16LeetCode每日一题——链表随机节点

最新推荐文章于 2022-12-11 16:23:45 发布

小唐要努力

最新推荐文章于 2022-12-11 16:23:45 发布

阅读量97

点赞数

分类专栏： # 力扣每日一题文章标签：链表

本文链接：https://blog.csdn.net/weixin_39478524/article/details/122521922

版权

力扣每日一题专栏收录该内容

80 篇文章 1 订阅

订阅专栏

链表随机节点

描述

给你一个单链表，随机选择链表的一个节点，并返回相应的节点值。

每个节点被选中的概率一样。

实现 Solution 类：

Solution(ListNode head) 使用整数数组初始化对象。
int getRandom() 从链表中随机选择一个节点并返回该节点的值。链表中所有节点被选中的概率相等。

示例

输入
["Solution", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom"]
[[[1, 2, 3]], [], [], [], [], []]

输出
[null, 1, 3, 2, 2, 3]

解释
Solution solution = new Solution([1, 2, 3]);
solution.getRandom(); // 返回 1
solution.getRandom(); // 返回 3
solution.getRandom(); // 返回 2
solution.getRandom(); // 返回 2
solution.getRandom(); // 返回 3
// getRandom() 方法应随机返回 1、2、3中的一个，每个元素被返回的概率相等。

提示

链表中的节点数在范围 $[1, 10^4]$ 内
$-10^4 \le Node.val \le 10^4$
至多调用 getRandom 方法 $10^4$ 次

进阶

如果链表非常大且长度未知，该怎么处理？
你能否在不使用额外空间的情况下解决此问题？

方法：记录所有链表元素

我们可以在初始化时，用一个数组记录链表中的所有元素，这样随机选择链表的一个节点，就变成在数组中随机选择一个元素。

class Solution {
    List<Integer> list;
    Random random;

    public Solution(ListNode head) {
        list = new ArrayList<Integer>();
        while (head != null) {
            list.add(head.val);
            head = head.next;
        }
        random = new Random();
    }

    public int getRandom() {
        return list.get(random.nextInt(list.size()));
    }
}