python - leetcode返回单链表任意节点，提示未声明？

Question

就是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了

为什么啊？ 谢谢，实在不懂， 出错的那个测试用例我也不明白，这么多中括号表示什么啊？

Answer 1

因為測試是對同一個Solution物件多次呼叫getRandom，你在出錯的版本中修改了物件的self.head，在多次調用後self.head為None，while循環未執行，從而報錯說res未定義；修改後的版本則沒有改變物件head，所以沒有出錯。

Answer 2

問題出在變數res上，getRandom()返回的是res，但变量res在返回之前，只在特定条件（满足while条件和if条件）下才被赋值。假设你的while循环体没有被执行，那么res在回傳前沒有被賦值，python解釋器不知道該回傳什麼值，所以才報錯。