一、首先二叉樹的定義:
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
? ? ? ? 構建一棵二叉樹:
class Node(object):
def __init__(self, val):
self.val = val
self.lchild = None
self.rchild = None
class Tree(object):
def __init__(self):
self.root = None
self.myQueue = []
def add(self, val):
node = Node(val)
if self.root == None:
self.root = node
self.myQueue.append(self.root)
else:
while True:
point = self.myQueue[0]
if point.lchild == None:
point.lchild = node
self.myQueue.append(point.lchild)
return
elif point.rchild == None:
point.rchild = node
self.myQueue.append(point.rchild)
self.myQueue.pop(0)
return
?
二、前序遍歷
遞歸實現
def preorder(root,res=[]):
if not root:
return
res.append(root.val)
preorder(root.lchild,res)
preorder(root.rchild,res)
return res
迭代實現
def preorder(root):
res = []
if not root:
return []
stack = [root]
while stack:
node = stack.pop()
res.append(node.val)
if node.rchild:
stack.append(node.rchild)
if node.lchild:
stack.append(node,lchild)
return res
? 三、中序遍歷
遞歸實現
def inorder(root,res=[]):
if not root:
return
inorder(root.lchild,res)
res.append(root.val)
inorder(root.rchild,res)
return res
迭代實現
def inorder(root):
stack = []
node = root
res = []
while stack or node:
while node:
stack.append(node)
node = node.lchild
node = stack.pop()
res.append(node.val)
node = node.rchild
return res
? 四、后序遍歷
遞歸實現
def laorder(root,res=[]):
if not root:
return
laorder(root.lchild,res)
laorder(root.rchild,res)
res.append(root.val)
return res
迭代實現
def laorder(root):
stack = [root]
res = []
while stack:
node = stack.pop()
if node.lchild:
stack.append(node.left)
if node.rchild:
stack.append(node.right)
res.append(node.val)
return res[::-1]
五、層次遍歷?
迭代
def levelorder(root):
queue = [root]
res = []
while queue:
node = queue.pop(0)
if node.lchild:
queue.append(node.lchild)
if node.rchild:
queue.append(node.rchild)
res.append(node.val)
return res
?運行結果:
?
?參考博客:https://www.jb51.net/article/157422.htm、https://blog.csdn.net/Bone_ACE/article/details/46718683、https://blog.csdn.net/wzngzaixiaomantou/article/details/81294915
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元
