2022.1.16 力扣-每日一题-链表随机节点

题目描述：

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

实现 Solution 类：

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

方法一：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    int n = 0;
    vector<int> lis;
    Solution(ListNode* head) {
        while (head != nullptr)
        {
            n++;
            lis.push_back(head->val);
            head = head->next;
        }
    }
    
    int getRandom() {
        int x = rand() % n;
        return lis[x];
    }
};

最简单的方法，初始化O(n)，随机O(1)

方法二（蓄水池抽样法）：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* head;
    Solution(ListNode* head) {
        this->head = head;
    }
    
    int getRandom() {
        ListNode* cur = head;
        int i = 0;
        int ans;
        while (cur != nullptr)
        {
            i++;
            if (rand() % i == 0)
            {
                ans = cur->val;
            }
            cur = cur->next;
        }
        return ans;
    }
};

新知识get，主要思路就是从前往后遍历到第i个结点时（i从1开始算起），如果在 [0，i) 范围内随机到0，就将答案设置为当前值，这样遍历完一次之后每个节点被选中的概率都是相等的1/n。至于证明还得用到数学推导，还是略过吧。