[leetcode]随机返回K个未知链表长度的节点值

原理为蓄水池抽样：一个未知大小的数据流，随机返回k个数据，每个样本被选中的概率是相同的。所以可以直接读入的时候，随机返回。
证明呢？

数学归纳法
假设前$N$条数据，返回$K$条，每条被返回的概率相同，都是$\frac{K}{N}$。
那么$N+1$条数据返回$K$条的概率，用$\frac{K}{N+1}$的概率去选新增的第$K+1$条，那么元素中被采样的概率为：
1、选中了新增的这一个，前N条中有一条被替换：$\frac{K}{N+1}\ast \frac{K-1}{K}\ast \frac{K}{N}=\frac{K-1}{N}\ast \frac{K}{N+1}$
2、第$N+1$条没有被选中：$(1-\frac{K}{N+1})\ast \frac{K}{N}=\frac{N+1-K}{N+1}\ast \frac{K}{N}$
概率相加结果为：$\frac{K}{N+1}$
结果得证。

代码为：
random() 结果返回实数，范围在[0,1)之内。

class Solution:
    def __init__(self, head: ListNode):
        """
        @param head The linked list's head.
        Note that the head is guaranteed to be not null, so it contains at least one node.
        """
        self.head = head
    def getRandom(self) -> int:
        """
        Returns a random node's value.
        """       
        count,res = 1,-1
        from random import random
        cur = self.head
        while(cur):
            if(int(random()*count) == 0):
                res = cur.val
            count, cur = count + 1, cur.next
        return res