382.链表随机节点

题目

382.链表随机节点

题目大意

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

实现 Solution 类：

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

样例

数据规模

思路1

可以在初始化时，用一个数组记录链表中的所有元素，这样随机选择链表的一个节点，就变成在数组中随机选择一个元素，相当于每次rand一个数∈[0,tot-1]，然后返回a[rand值]。

代码1

class Solution {
public:
    int a[100000+50],tot;
    Solution(ListNode* head) {
        while(head){
            a[tot++]=head->val;
            head=head->next;
        }
    }
    
    int getRandom() {
        int x=rand()%tot;
        return a[x];
    }
};

思路2

方法一需要花费 $O(n)$ 的空间存储链表中的所有元素，那么能否做到 $O(1)$ 的空间复杂度呢？

可以设计算法：从链表头开始，遍历整个链表，对遍历到的第$i$个节点，随机选择区间$[0,i)$内的一个整数，如果其等于$0$，则将答案置为该节点值，否则答案不变。

该算法会保证每个节点的值成为最后被返回的值的概率均为$\frac{1}{n}$

代码2

class Solution {
public:
    ListNode* a;
    Solution(ListNode* head) {
        a=head;
    }
    
    int getRandom() {
        int i=1,ans=0;
        for(auto Node=a;Node;Node=Node->next){
            if(rand()%i==0){
                ans=Node->val;
            }
            i++;
        }
        return ans;
    }
};