-
Notifications
You must be signed in to change notification settings - Fork 1
/
382.cpp
36 lines (33 loc) · 928 Bytes
/
382.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
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 *head;
Solution(ListNode* _head): head(_head) {}
/** Returns a random node's value. */
int getRandom() {
int selected = head->val, count = 2;
ListNode *p = head->next;
while (p != nullptr) {
if ((int)((double)rand() / (RAND_MAX + 1u) * count) == 0) {
selected = p->val;
}
p = p->next;
count++;
}
return selected;
}
};
/**
* Your Solution object will be instantiated and called as such:
* Solution obj = new Solution(head);
* int param_1 = obj.getRandom();
*/