python并行赋值3个变量
我的问题来自一个流行的编码测试。
给定一个定义如下的链:
class ListNode:
def __init__(self, x):
self.val = x
self.next = None
如果我们想恢复链条,例如:
输入:1->2->3->4->5->NULL
输出:5->4->3->2->1->NULL
它可以解决像这样:
class Solution:
def reverseList(self, head: ListNode) -> ListNode:
cur, pre = head, None
while cur:
tmp = cur.next
cur.next = pre
pre = cur
cur = tmp
return pre
但是有一种更紧凑的方式,我无法真正遵循:
class Solution:
def reverseList(self, head: ListNode) -> ListNode:
cur, pre = head, None
while cur:
cur.next, pre, cur = pre, cur, cur.next
return pre
因为如果我将并行分配行更改为
pre, cur, cur.next = cur, cur.next, pre
它不会再正常工作了。
我想知道 python 的并行分配是如何工作的,尤其是在所有 3 个变量都是动态的情况下。
回答
当你写平行作业时
x, y, z = a, b, c
它相当于
temp = (a, b, c)
x = temp[0]
y = temp[1]
z = temp[2]
所以在失败的版本中,它相当于
temp = (cur, cur.next, pre)
pre = temp[0]
cur = temp[1]
cur.next = temp[2]
将此与工作版本进行比较:
cur.next = temp[0]
pre = temp[1]
cur = temp[2]
不同之处在于,在您的版本中,您在步进cur.next
后分配cur
给cur.next
,因此您实际上是分配给原始cur.next.next
.