leetCode 382 链表的随机节点 （水库采样算法）

题目链接：点击查看

题目描述：

给定一个单向链表，要求设计一个算法，可以随机取得其中的一个数字。

输入输出：
 

输入： 

 

1->2->3->4->5

 

输出： 

 

3（在这个样例中，我们有均等的概率得到任意一个节点，比如 3 ）

      

题目分析：
 

不同于数组，在未遍历完链表前，我们无法知道链表的总长度。这里我们就可以使用水库采样：遍历一次链表，在遍历到第 m 个节点时，有 1/m 的概率选择这个节点覆盖掉之前的节点选择。详见如下代码。

 

代码：

class Solution
{
	ListNode *head;
public:
	Solution(ListNode*n):head(n){};
	int getRandom()
	{
		int ans=head->val;
		ListNode *node=head->next;
		int i=2;
		while(node)
		{
			if((rand()%i)==0)//1/i的概率生成随机数0 模拟水库算法中要求的1/i概率 
			{
				ans=node->val;
			}
			++i;
			node=node->next;
		}
		return ans;
	}
};