力扣382题 链表随机节点

382. 链表随机节点

给你一个单链表，随机选择链表的一个节点，并返回相应的节点值。每个节点 被选中的概率一样 。

实现 Solution 类：

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

 

提示：

• 链表中的节点数在范围 [1, 104] 内
• -104 <= Node.val <= 104
• 至多调用 getRandom 方法 104 次

解题思路：① list存储每个结点的值，然后使用 random 类来完成等概率选取。② 随机池塘抽样

① 代码实现如下：

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()));
    }
}

② 随机池塘抽样 代码实现如下：

class Solution {
    ListNode head;
    Random random;

    public Solution(ListNode head) {
        this.head = head;
        random = new Random();
    }

    public int getRandom() {
        int i = 1, ans = 0;
        for (ListNode node = head; node != null; node = node.next) {
            if (random.nextInt(i) == 0) { // 1/i 的概率选中（替换为答案）
                ans = node.val;
            }
            ++i;
        }
        return ans;
    }
}

总结：随机池塘抽样方法的证明，该算法会保证每个节点的值成为最后被返回的值的概率均为 1/n，证明如下：

 举个例子：① 如果只有一个结点，那么 [0,1) 随即返回取到 0 的概率是百分之百，因为整数只有一个 0。② 如果有两个结点，那么第一个结点被选中返回的概率 = 第一个中（1）加上第二个结点不中（1/2） = 1 * 1/2 = 1/2。第二个结点被选中返回的概率 = [0,2) 中随机整数到 0 的概率，也就是 1/2。③ 如果有三个结点，那么第一个结点被选中返回的概率 = 第一个中（1）加上第二个结点不中（1/2）加上第三个不中（2/3） = 1 * 1/2 * 2/3 = 1/3。第二个结点被选中返回的概率 = 第二个中（1/2）加上第三个不中（2/3） = 1/2 * 2/3 = 1/3 。第二个结点被选中返回的概率 = [0,3) 中随机整数到 0 的概率，也就是 1/3。