[leetcode] 382. Linked List Random Node
来源:互联网 发布:容迟网络 编辑:程序博客网 时间:2024/05/16 11:02
Given a singly linked list, return a random node's value from the linked list. Each node must have the same probability of being chosen.
Follow up:
What if the linked list is extremely large and its length is unknown to you? Could you solve this efficiently without using extra space?
Example:
// Init a singly linked list [1,2,3].ListNode head = new ListNode(1);head.next = new ListNode(2);head.next.next = new ListNode(3);Solution solution = new Solution(head);// getRandom() should return either 1, 2, or 3 randomly. Each element should have equal probability of returning.solution.getRandom();
这道题是在未知长度链表中等概率的随机取节点,题目难度为Medium。
由于每次取随机节点时不知道链表长度,所以不能简单用rand()函数来取随机节点。这里介绍下蓄水池抽样的基本思路,了解的同学可以跳过。假定最终选出的节点为node,第一次直接选第一个节点;第二次以1/2的概率决定是否拿第二个节点替换node;第三次以1/3的概率决定是否拿第三个节点替换node;以此类推,直至链表最后一个节点。这种方法能够保证所有节点被选中的概率均为1/n(n为当前节点数),即保证了等概率的随机选取。证明如下:
对于第m个节点,被选中的概率为:它被选中的概率 * 之后节点都没被选中的概率,即(1/m) * (m/(m+1)) * ((m+1)/(m+2)) * ... * ((n-1)/n) = 1/n.
具体代码:
class Solution { ListNode* hd;public: /** @param head The linked list's head. Note that the head is guanranteed to be not null, so it contains at least one node. */ Solution(ListNode* head) : hd(head) {} /** Returns a random node's value. */ int getRandom() { ListNode* node = hd; int cnt = 0, randVal = 0; while(node) { if(rand() % (++cnt) == 0) randVal = node->val; node = node->next; } return randVal; }};/** * Your Solution object will be instantiated and called as such: * Solution obj = new Solution(head); * int param_1 = obj.getRandom(); */
还可以将问题扩展到在未知或大样本中等概率随机取k个节点,思路是一样的。首先选出前k个节点放入蓄水池;第k+1个节点以k/(k+1)的概率决定是否将其换入蓄水池,如果换入则随机选择k个节点中的一个进行替换;同理第k+2个节点以k/(k+2)的概率决定是否将其换入蓄水池;直至最后节点,最终能够保证所有节点被选中的概率均为k/n,证明如下:
对于第m个节点,被选中的概率为:它被选中的概率 * (之后节点没被选中的概率 + 之后节点被选中但没有替换掉它的概率),即(k/m) * ((m+1-k)/(m+1) + (k/(m+1)) * ((k-1)/k)) * ... * ((n-k)/n + (k/n) * ((k-1)/k)) = k/n.
wiki上蓄水池抽样的页面,感兴趣的同学可以看下:https://en.wikipedia.org/wiki/Reservoir_sampling
- [leetcode] 382. Linked List Random Node
- <LeetCode OJ> 382. Linked List Random Node
- LeetCode—382. Linked List Random Node
- 【leetcode】382. Linked List Random Node
- [leetcode]382. Linked List Random Node
- Leetcode 382. Linked List Random Node
- [LeetCode]382. Linked List Random Node
- LeetCode 382. Linked List Random Node
- [leetcode]382. Linked List Random Node
- [Leetcode]382. Linked List Random Node
- 【LeetCode】 382. Linked List Random Node
- [leetcode]382. Linked List Random Node
- LeetCode 382. Linked List Random Node
- [LeetCode]382. Linked List Random Node
- leetcode 382. Linked List Random Node
- [LeetCode] 382. Linked List Random Node
- LeetCode 382. Linked List Random Node
- leetcode 382. Linked List Random Node
- Java集合容器优化
- Java HashSet详解
- python协程
- java中关于ArrayList,LinkedList,HashSet,Vector,TreeSet的区别和使用
- 【WD4A学习笔记】- web dynpro for abap 在线帮助文档
- [leetcode] 382. Linked List Random Node
- C# 操作IIS方法集合
- CSS3 3D动画效果
- 从精益软件到精益思想
- 一起读guavaCache源码
- 博弈 SG函数
- MySQL---数据库从入门走向大神系列(十)-Connection对象池、装饰模式与动态代理模式
- 使用android studio 连接海马玩模拟器
- MySQL使用SELECT INTO OUTFILE导出文本文件