Leetcode—382. 链表随机节点【中等】（水塘抽样法）

2024每日刷题（一零九）

Leetcode—382. 链表随机节点

在这里插入图片描述

算法思想

我们可以在初始化时，用一个数组记录链表中的所有元素，这样随机选择链表的一个节点，就变成在数组中随机选择一个元素

实现代码

/*** 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:vector<int> arr;Solution(ListNode* head) {for(ListNode* cur = head; cur; cur=cur->next) {arr.push_back(cur->val);}}int getRandom() {return arr[rand()%arr.size()];}
};/*** Your Solution object will be instantiated and called as such:* Solution* obj = new Solution(head);* int param_1 = obj->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:Solution(ListNode* head): sohead(head) {}int getRandom() {int ans = 1e5;int i = 1;for(ListNode* cur = sohead; cur; cur = cur->next, i++) {if(rand()%i==0) {ans = cur->val;}}return ans;}
private:ListNode* sohead;
};/*** Your Solution object will be instantiated and called as such:* Solution* obj = new Solution(head);* int param_1 = obj->getRandom();*/