Linked List Random Node

水塘抽样算法:

关键点是不知道样本总大小,抽样要等概率

算法过程:遍历一遍链表,每遍历到一个元素时记当前元素编号为n,生成随机数,若随机数小于1/n,则令该元素作为结果,这样进行下去,遍历到最后得到的结果的被抽取概率就是1/N,也就能保证所有元素都是等概率抽取的。

参考公式为:

                        

class Solution {
public:
	/** @param head The linked list's head.
	Note that the head is guaranteed to be not null, so it contains at least one node. */
	ListNode* p;
	Solution(ListNode* head) {
		p = head;
	}

	/** Returns a random node's value. */
	int getRandom() {
		int res = p->val;
		ListNode* node = p->next;
		int i = 2;
		while (node) {
			int j = rand() % i;
			if (j == 0)//这样写的原理是:若生成的随机数在0~i之间,则j=0时rand()=i,
                                   //若生成的随机数在0~2i之间时,rand()=i或2i时j=0,以此类推,这样就保证了上面说明中随机数小于1/n那条性质
				res = node->val;
			i++;
			node = node->next;
		}
		return res;
	}
};


阅读更多
个人分类: leetcode
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭