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

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