樹的前序(preorder)遍歷:
iterative:
用stack界阁,依次放右節(jié)點侯繁,左節(jié)點
python代碼:
class Solution:
"""
@param root: A Tree
@return: Preorder in ArrayList which contains node values.
"""
def preorderTraversal(self, root):
# write your code here
res = []
stack = []
if not root:
return res
stack.append(root)
while stack:
node = stack.pop()
res.append(node.val)
if node.right:
stack.append(node.right)
if node.left:
stack.append(node.left)
return res
recursive:
就很簡單啦,res添加當前節(jié)點泡躯,dfs訪問左節(jié)點贮竟,dfs訪問右節(jié)點。
class Solution:
"""
@param root: A Tree
@return: Preorder in ArrayList which contains node values.
"""
def preorderTraversal(self, root):
# write your code here
res = []
self.dfs(root, res)
return res
def dfs(self, node, res):
if node:
res.append(node.val)
self.dfs(node.left, res)
self.dfs(node.right, res)
return
樹的中序遍歷(inorder): 左根右
iterative:
同樣是用stack较剃,從根開始依次把左邊節(jié)點往棧里推坝锰,直到節(jié)點為空,棧pop重付,放入result list中顷级,再把當前節(jié)點設置為pop出的節(jié)點的右節(jié)點,再執(zhí)行上述操作确垫。最后直到棧為空
"""
Definition of TreeNode:
class TreeNode:
def __init__(self, val):
self.val = val
self.left, self.right = None, None
"""
class Solution:
"""
@param root: A Tree
@return: Inorder in ArrayList which contains node values.
"""
def inorderTraversal(self, root):
# write your code here
stack = []
res = []
if not root:
return []
cur = root
while cur or stack:
while cur:
stack.append(cur)
cur = cur.left
node = stack.pop()
res.append(node.val)
cur = node.right
return res
recursive:
class Solution:
"""
@param root: A Tree
@return: Inorder in ArrayList which contains node values.
"""
def inorderTraversal(self, root):
# write your code here
res = []
self.dfs(root, res)
return res
def dfs(self, node, res):
if node:
self.dfs(node.left, res)
res.append(node.val)
self.dfs(node.right, res)
樹的后序遍歷(post order):
iterative:
class Solution:
"""
@param root: A Tree
@return: Postorder in ArrayList which contains node values.
"""
def postorderTraversal(self, root):
# write your code here
res = []
stack = []
if not root:
return res
stack.append(root)
while stack:
node = stack.pop()
res.append(node.val)
if node.left:
stack.append(node.left)
if node.right:
stack.append(node.right)
res.reverse()
return res