轉(zhuǎn)換函數(shù)將二叉樹轉(zhuǎn)換為鏈表。由于使用全局變量，我無法通過 geeksforgeek 中的測(cè)試用例。如何避免使用全局變量？ prev = None head = Nonedef convert(root): global prev global head if root is None: return convert(root.left) if head== None: head = root prev = root else: root.left = prev prev.right = root prev = root convert(root.right)
2 回答
德瑪西亞99
TA貢獻(xiàn)1770條經(jīng)驗(yàn) 獲得超3個(gè)贊
將它們作為參數(shù)傳遞，默認(rèn) = None：
def convert(root, prev = None, head = None):
if root is None:
return
convert(root.left, prev=prev, head=head)
if head== None:
head = root
prev = root
else:
root.left = prev
prev.right = root
prev = root
convert(root.right, prev=prev, head=head)
holdtom
TA貢獻(xiàn)1805條經(jīng)驗(yàn) 獲得超10個(gè)贊
我認(rèn)為你需要返回鏈表的頭。
不管怎樣，你可以包裝你的函數(shù)來為這兩個(gè)變量創(chuàng)建一個(gè)閉包：
def convert(root):
head = None
prev = None
def recur(cur):
nonlocal prev, head
if cur is None:
return
recur(cur.left)
if head is None:
head = cur
else:
cur.left = prev
prev.right = cur
prev = cur
recur(cur.right)
recur(root)
return head
添加回答
舉報(bào)
0/150
提交
取消