链表随机节点的思路探讨与源码
链表随机节点的题目如下图,该题属于链表类和数学类型的题目,主要考察对于数学方法的使用和链表结构的理解。本文的题目作者想到2种方法,分别是记录链表方法和水塘抽样方法,其中记录链表方法使用Java进行编写,而水塘抽样方法使用Python进行编写,当然这可能不是最优的解法,还希望各位大佬给出更快的算法。
本人认为该题目可以使用记录链表方法的思路进行解决,首先初始化参数,然后在列表初始化的时候,判断每一个列表的节点,这些节点是否非空,如果是就将链表的结果添加到列表里,在循环结束后,对随机数类进行初始化。对列表的取值的时候,从随机数实例里随机找一个值进行返回。那么按照这个思路我们的Java代码如下:
#喷火龙与水箭龟
class Solution {
Random random;
List<Integer> arrayList;
public Solution(ListNode node) {
arrayList = new ArrayList<Integer>();
while (node != null) {
arrayList.add(node.val);
node = node.next;
}
random = new Random();
}
public int getRandom() {
return arrayList.get(random.nextInt(arrayList.size()));
}
}
显然,我们的记录链表方法的效果还不错,还可以使用水塘抽样的方法解决。首先将链表的首个元素取出并进行参数的初始化,然后开始遍历循环链表,判断随机选择的数是否等于0,如果是就把链表的值赋值给结果参数,否则就继续循环。直到最终遍历结束并返回结果参数。所以按照这个思路就可以解决,下面是Python代码:
#喷火龙与水箭龟
class Solution:
def __init__(self, head: Optional[ListNode]):
self.head = head
def getRandom(self) -> int:
node= self.head
ir = 1
res = 0
while node:
if(randrange(ir) == 0):
res = node.val
ir = ir + 1
node = node.next
return res
从结果来说Java版本记录链表方法的效率不错,而Python版本的水塘抽样方法的速度比较一般,但应该是有更多的方法可以进一步提速的,希望朋友们能够多多指教,非常感谢。