就是leetcode382
Linked List Random Node
Given a singly linked list, return a random node's value from the linked list. Each node must have the same probability of being chosen.
然后我写的问题如下:
但是把self.head用新变量head,s或者其他什么字母声明一下,改成:
···
cnt = 0
head = self.head
while head:
if random.randint(0, cnt) == 0:
ans = head.val
head = head.next
cnt += 1
return ans
就通过了,AC了
为什么啊? 谢谢,实在不懂, 出错的那个测试用例我也不明白,这么多中括号表示什么啊?
因为测试是对同一个Solution对象多次调用getRandom,你在出错的版本中修改了对象的self.head,在多次调用后self.head为None,while循环未执行,从而报错说res未定义;修改的版本则没有改变对象head,所以没有出错。
问题出在变量
res
上,getRandom()
返回的是res
,但变量res
在返回之前,只在特定条件(满足while
条件和if
条件)下才被赋值。假设你的while
循环体没有被执行,那么res
在返回前没有被赋值,python解释器不知道该返回什么值,所以才报错。