Python:樹的遍歷

各類遍歷順序以下圖所示:app

樹的最大深度 spa

# class TreeNode(object): # def __init__(self, x): # self.val = x # self.left = None # self.right = None

class Solution(object): def maxdepth(self, root): if root is None: return 0 return max(self.maxdepth(root.left), self.maxdepth(root.right))+1

 

深度優先code

深度優先遍歷有三種方式:前序遍歷、中序遍歷和後序遍歷blog

所說的前序、中序、後序,是指根節點的前後順序。it

前序遍歷:根節點 -> 左子樹 -> 右子樹io

# class TreeNode(object): # def __init__(self, x): # self.val = x # self.left = None # self.right = None

class Solution(object): def preorder(self, root): if root is None: return ''
print root.val if root.lef: self.preorder(root.left) if root.right: self.preorder(root.right)

中序遍歷:左子樹 -> 根節點 -> 右子樹class

# class TreeNode(object): # def __init__(self, x): # self.val = x # self.left = None # self.right = None

class Solution(object): def midorder(self, root): if root is None: return ''
        if root.lef: self.midorder(root.left) print root.val if root.right: self.midorder(root.right)

後序遍歷:左子樹 -> 右子樹 -> 根節點object

# class TreeNode(object): # def __init__(self, x): # self.val = x # self.left = None # self.right = None

class Solution(object): def endorder(self, root): if root is None: return ''
        if root.lef: self.endorder(root.left) if root.right: self.endorder(root.right) print root.val

 

 廣度優先遍歷

廣度優先遍歷,即層次遍歷,優先遍歷兄弟節點queue

層次遍歷:根節點 -> 左節點 -> 右節點

# class TreeNode(object): # def __init__(self, x): # self.val = x # self.left = None # self.right = None

class Solution(object):   def graorder(self, root):     if root is None:       return ''     queue = [root]     while queue:       res = []       for item in queue:         print item.val,         if item.left:           res.append(item.left)         if item.right:           res.apppend(item.right)       queue = res

 

比較兩棵樹是否相同

# class TreeNode(object): # def __init__(self, x): # self.val = x # self.left = None # self.right = None

class Solution(object): def issame(self, root1, root2): if root1 is None and root2 is None: return True elif root1 and root2: return root1.val==root2.val and issame(root1.left, root2.left) and issame(root1.right, root2.right) else: return False
相關文章
相關標籤/搜索