JAVA程序设计：链表随机节点（LeetCode：382）

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

进阶:
如果链表十分大且长度未知，如何解决这个问题？你能否使用常数级空间复杂度实现？

示例:

// 初始化一个单链表 [1,2,3].
ListNode head = new ListNode(1);
head.next = new ListNode(2);
head.next.next = new ListNode(3);
Solution solution = new Solution(head);

// getRandom()方法应随机返回1,2,3中的一个，保证每个元素被返回的概率相等。
solution.getRandom();

思路：蓄水池抽样法

从左向右遍历链表，每遇到一个节点，计数器cnt加1。然后生成随机数j在范围[0,…,cnt-1]中。如果j < 1 那么，取此位置。如果不是，则不取。

概率证明：第cnt个节点被选中的概率为1/cnt。蓄水池中被选中要替换的节点的概率为1/1。综合看第cnt个节点在蓄水池中的概率为1/cnt * 1/1 = 1/cnt。

数学归纳法：

初步，cnt = 1时，被选中的概率1/1=1;

归纳，假设在cnt-1时，节点被选中在蓄水池中的概率是1/(cnt-1)。由上结果，在cnt时，蓄水池中节点被替换的概率是1/cnt。那么，不被替换的概率是(cnt-1)/cnt。因此cnt-1时，节点保留在蓄水池中的概率为1/(cnt-1) * (cnt-1)/cnt = 1/cnt。

因此，第cnt个节点在蓄水池中的概率为1/cnt，而前cnt-1个节点保留在蓄水池中的概率也为1/cnt。如此保证等概率。

class Solution {

    private ListNode node;
    
    public Solution(ListNode head) {
        node = head;
    }
    
    public int getRandom() {
        ListNode res = node;
        ListNode cur = node.next;
        int i = 2;
        //从第二个节点开始,每次循环替换res的概率都是1/i
        while(cur != null){
            Random random = new Random();
            int ran = random.nextInt(i);
            if(ran == 0){
                res = cur;
            }
            cur = cur.next;
            i++;
        }
        return res.val;
    }
}