LeetCode第 382 题：链表随机节点(C++)

382. 链表随机节点 - 力扣（LeetCode）

注意：要求保证每个节点被选取的概率是一样的。

普通解

先进行遍历，得到节点总数，然后产生随机数，计数返回就行，一共遍历了两次。

class Solution {
public:
    /** @param head The linked list's head.
        Note that the head is guaranteed to be not null, so it contains at least one node. */
    Solution(ListNode* head) {
        this->head = head;
        auto p = head;
        while(p){
            ++len;
            p = p->next;
        }
    }

    int getRandom() {
        int i = rand()%len;//0～len-1之间的随机数
        auto cur = head;
        while(i){
            cur = cur->next;
            --i;
        }
        return cur->val;
    }
private:
    ListNode *head;
    int len = 0;
};

进阶

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

这道题的普通解很简单，但是重点是进阶思路这儿如何实现。

这是典型的蓄水池抽样问题：

蓄水池抽样——《编程珠玑》读书笔记

意味着我们不能一直遍历下去直到得知链表的长度，然后去产生随机数。

正确的解法是：

先选择第一个对象，然后以1/2的概率选择第二个对象（如果选择了第二个对象，第一个自然就舍弃了），再以1/3的概率选择第三个对象…以此类推，以1/m的概率选择第m个对象。最后选择过程结束时，每一个对象被选中的概率就是 1/n(n为对象总数)。

ok，虽然结论有了，但是怎么证明每个对象被选中的概率就是 1/n 呢：

第m个对象被选中的概率 p = 选择它的概率 * 它后面的所有对象均不被选择的概率，也就是

$$P=\frac{1}{m}\times \left(\frac{m}{m+1}\times \frac{m+1}{m+2}\times \dots \times \frac{n-1}{n}\right)=\frac{1}{n}$$

class Solution {
public:
    /** @param head The linked list's head.
        Note that the head is guaranteed to be not null, so it contains at least one node. */
    Solution(ListNode* head) : head(head){}
    
    /** Returns a random node's value. */
    int getRandom() {
        auto cur = head;
        int cnt = 1, res = cur->val;
        while(cur){
            if(rand()%cnt == 0) res = cur->val;
            ++cnt;
            cur = cur->next;
        }
        return res;
    }
private:
    ListNode* head;
};

如果抽取的不止一个数，有k个数怎么办呢？

蓄水池抽样算法 傻子都能看懂 - 链表随机节点 - 力扣（LeetCode）

比着推导一遍就懂了，过程很简单。