未知长度链表中随机取出其中某一节点的值

转自：

 

题目描述：有一个单向链表的长度未知，怎样从中随机取出一个节点？要求每个节点被选中的概率相等。

解决方法：

   遍历一次链表，用一个临时变量pTemp指向返回的节点，设一个计数器iCount统计已遍历的节点个数，然后生成0到iCount-1之间的随机数；若生成的随机数为0，则将pTemp指针替换为当然节点的地址。可以证明对每一个节点，它被选中的概率为1/n，n为链表的长度。

    对于第i个节点，被选中的概率为:p = 1/i *(i  /i+1)*(i+1  /  i+2)*......(n-1  / n) = 1/n

代码如下：

 

[cpp]  view plain copy

 

1. #include<iostream>  
2. #include<cstring>  
3. #include<cmath>  
4. using namespace std;  
5.   
6. struct Node  
7. {  
8.     int iVal;  
9.     Node* next;  
10. };  
11.   
12. Node* GetOneNode(Node* head)  
13. {  
14.     if(head == NULL)  
15.         return NULL;  
16.     int iCount = 1;  
17.     Node* tmp = head->next;  
18.     Node* pRet = tmp;  
19.     while(tmp)  
20.     {  
21.         int ran = rand();  
22.         int t = ran%iCount;//生成0到iCount-1之间的一个随机数  
23.         if(t == 0)  
24.             pRet = tmp;  
25.         iCount++;  
26.         tmp = tmp->next;  
27.     }  
28.     return pRet;  
29. }  
30.   
31. Node* initLinkList(Node* head, int N)  
32. {  
33.     Node* p = NULL;  
34.     Node* pTemp = head;  
35.     for(int i=0; i<N; i++)  
36.     {  
37.         p = new Node;  
38.         p->iVal = rand();  
39.         p->next = NULL;  
40.         pTemp->next = p;  
41.         pTemp = p;  
42.     }  
43.     return head;  
44. }  
45.   
46. bool freeList(Node* head)  
47. {  
48.     Node* p = head;  
49.     while(head)  
50.     {  
51.         p = head->next;  
52.         delete head;  
53.         head = p;  
54.     }  
55.     return true;  
56. }  
57.   
58. int main()  
59. {  
60.     Node* head = new Node;  
61.     head = initLinkList(head, 10);  
62.     Node* pRet = GetOneNode(head);  
63.     cout<<pRet->iVal<<endl;  
64.     freeList(head);  
65.     return 0;  
66. }