114 二叉樹展開為鍊錶

一開始寫的：

#
## [114] 二叉樹展開為鍊錶
## definition for a binary tree node.
# class treenode:
#     def __init__(self, x):
#         self.val = x
#         self.left = none
#         self.right = none
class solution:
def flatten(self, root: treenode) -> none:
"""do not return anything, modify root in-place instead.
"""def preorder(node):
if not node:
return
preorder(node.left)
preorder(node.right)
if not root:
return
temp=
preorder(root)
for i in range(len(temp)-1):
temp[i].left=none
temp[i].right=temp[i+1]
temp[len(temp)-1].left=none
temp[len(temp)-1].right=none

能通過，但看別人的題解發現題幹寫了「原地」，我這個申請了額外的陣列儲存前序遍歷的節點顯然不對了。然後看別人的後序做法又自己寫。乙個關鍵問題，為什麼不能用中序或前序：因為題幹最後要求的順序是前序：即中，左，右，在考察某特定節點時，將其左右子樹全部處理完畢之前不能將該節點與其左右子樹的連線改掉（想了半天），所以只有後序遍歷的順序是：左，右，中，在遍歷到中節點的時候，左右子樹已經處理好，此時可以改動中節點與左右子樹的連線了。

下面是參考題解自己寫的後序**，思路就是先處理左右子樹，然後處理根節點。根節點右指標連線左子樹最右元素p（即前序前繼），p右指標再連線根節點原右子樹，這樣就是中，左，右的前序順序了。

# definition for a binary tree node.
# class treenode:
#     def __init__(self, x):
#         self.val = x
#         self.left = none
#         self.right = none
class solution:
def flatten(self, root: treenode) -> none:
"""do not return anything, modify root in-place instead.
"""def func(node):
if not node:
return
func(node.left)
func(node.right)
p=node.left
while p and p.right:
p=p.right
if p:
node_right=node.right
node.right=node.left
node.left=none
p.right=node_right
func(root)

又寫了個迭代的：先處理右子樹成一條鍊錶，再處理左子樹成一條鍊錶，然後和根節點連起來。。

class solution:
def flatten(self, root: treenode) -> none:
"""do not return anything, modify root in-place instead.
"""if not root:
return
le,ri=none,none
if root.left:
le=root.left
if root.right:
ri=root.right
while ri and (ri.left or ri.right):
if not ri.left:
ri=ri.right
else:
p=ri.left
while p.right:
p=p.right
ri_right=ri.right
ri.right=ri.left
p.right=ri_right
ri.left=none
ri=ri.right     
while le and (le.left or le.right):
if not le.left:
le=le.right
else:
p=le.left
while p.right:
p=p.right
le_right=le.right
le.right=le.left
p.right=le_right
le.left=none
le=le.right
if le:
le.right=root.right
root.right=root.left
root.left=none

然後我發現：我為啥要分左右子樹再串起來？？明明從根節點乙個while迴圈就行了，腦子抽了！！

class solution:
def flatten(self, root: treenode) -> none:
"""do not return anything, modify root in-place instead.
"""if not root:
return
x=root
while x and (x.left or x.right):
if not x.left:
x=x.right
else:
p=x.left
while p.right:
p=p.right
x_right=x.right
x.right=x.left
p.right=x_right
x.left=none
x=x.right

114 二叉樹展開為鍊錶

114 二叉樹展開為鍊錶

114 二叉樹展開為鍊錶

114 二叉樹展開為鍊錶

相關推薦