(leetcode算法题)382. 链表随机节点

如果给你一个 智能记录 k行内容的小笔记本,从一本你也不知道有多少行的 C++ Primer中进行摘抄,你应该怎么做才能让抄写的时候能让书中的每一行都等概率的出现在小笔记本中?

答:准备好一个公平的轮盘和一个巨大的摇奖机,

轮盘上有k个数字,每个数字对应小笔记本的一行,抽中每个数字的概率相同

摇奖机装置可以从放入其中的 i 个纸条中等概率的抽出一个纸条

开始遍历 C++ Primer中的每一行,对于前k行,直接将他们的行号写到纸条上,放入摇奖机中

并且在一个record数组中记录下这些行号

在k行之后,同样将行号写到纸条上,然后把纸条放到摇奖机中开始摇奖,如果摇出来的纸条是轮盘上的某个数字num,那么将这个新放入的纸条上的行号替换调record中记录的这个数字,

直到把 C++ Primer中的所有行都遍历完,就能达到效果,

用数学归纳法就可以证明所有行都可以等概率的出现在小笔记本中,而且这个概率是 k / n

其中n 是 C++ Primer中的行数

假设读到 i - 1行,每一行能够出现在小笔记本上的概率是 k / (i - 1)

那么读到 i - 1 行时,第一行能够进入小笔记本的概率是

k / (i - 1) * (1 - (k / i) * (1 / k)) = k / i

图片来自B站UP主五点七边的视频【经典算法题】蓄水池抽样算法_哔哩哔哩_bilibili

382的代码

cpp 复制代码
/**
 * 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 {
ListNode* _head;
public:
    Solution(ListNode* head) {
        _head = head;
    }
    
    int getRandom() {
        int val = _head->val;
        ListNode* nextnode = _head->next;
        int count = 2;
        while(nextnode){
            int x = rand() % count;
            if(x == 0){
                val = nextnode->val;
            }
            count++;
            nextnode = nextnode->next;
        }
        return val;
    }
};

/**
 * Your Solution object will be instantiated and called as such:
 * Solution* obj = new Solution(head);
 * int param_1 = obj->getRandom();
 */
相关推荐
jndingxin1 小时前
OpenCV CUDA模块光流计算-----实现Farneback光流算法的类cv::cuda::FarnebackOpticalFlow
人工智能·opencv·算法
编程绿豆侠1 小时前
力扣HOT100之栈:394. 字符串解码
java·算法·leetcode
朝朝又沐沐1 小时前
基于算法竞赛的c++编程(18)string类细节问题
开发语言·c++·算法
记得早睡~2 小时前
leetcode73-矩阵置零
数据结构·leetcode·矩阵
爱coding的橙子2 小时前
每日算法刷题Day27 6.9:leetcode二分答案2道题,用时1h20min
算法·leetcode·职场和发展
GalaxyPokemon2 小时前
LeetCode - 3. 无重复字符的最长子串
算法·哈希算法·散列表
a.3023 小时前
C++ 时间处理指南:深入剖析<ctime>库
数据结构·c++·算法
亮亮爱刷题3 小时前
算法刷题-回溯
算法
Neil今天也要学习3 小时前
永磁同步电机无速度算法--自适应龙贝格观测器
算法
算AI5 小时前
AI辅助编程:常用的7种Prompt模式
人工智能·算法