数据结构
网络
RDBMS
操作系统
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 编程
C++
C#
MongoDB
MySQL
Javascript
PHP
物理
化学
生物
数学
英语
经济
心理学
社会研究
时装研究
法律研究在 C++ 中链接列表随机节点
假设我们有一个单链表，我们必须从链表中查找一个随机节点的值。此处每个节点被选中的概率必须相同。因此，例如，如果链表为 [1,2,3]，则它可以在 1、2 和 3 范围内返回随机节点。
为了解决此问题，我们将遵循以下步骤 −
在 getRandom() 方法中，执行以下操作 −
ret := -1, len := 1, v := x
while v 不为 null
如果 rand() 可被 len 整除，则 ret := v 的 val
将 len 增加 1
v := v 的 next
返回 ret
示例(C++)
让我们看以下实现来获得更好的理解 −
#include <bits/stdc++.h>
using namespace std;
class ListNode{
public:
int val;
ListNode *next;
ListNode(int data){
val = data;
next = NULL;
}
};
ListNode *make_list(vector<int> v){
ListNode *head = new ListNode(v[0]);
for(int i = 1; i<v.size(); i++){
ListNode *ptr = head;
while(ptr->next != NULL){
ptr = ptr->next;
}
ptr->next = new ListNode(v[i]);
}
return head;
}
class Solution {
public:
ListNode* x;
Solution(ListNode* head) {
srand(time(NULL));
x = head;
}
int getRandom() {
int ret = -1;
int len = 1;
ListNode* v = x;
while(v){
if(rand() % len == 0){
ret = v->val;
}
len++;
v = v->next;
}
return ret;
}
};
main(){
vector<int> v = {1,7,4,9,2,5};
ListNode *head = make_list(v);
Solution ob(head);
cout << (ob.getRandom());
}输入
Initialize list with [1,7,4,9,2,5] Call getRandom() to get random nodes
输出
4 9 1
更新于: 02-May-2020
320 次浏览
广告