二叉树层次遍历python_二叉树遍历python3代码(先序、中序、后序、层次)(递归、非递归)...

# Definition for a binary tree node.

# class TreeNode:

# def __init__(self, x):

# self.val = x

# self.left = None

# self.right = None

(一)二叉树的中序遍历

递归:

class Solution:

def inorderTraversal(self, root: TreeNode) -> List[int]:

res=[]

if root:

res+=self.inorderTraversal(root.left)

res.append(root.val)

res+=self.inorderTraversal(root.right)

return res

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

class Solution:

def inorderTraversal(self, root: TreeNode) -> List[int]:

if not root:

return []

return self.inorderTraversal(root.left) + [root.val] + self.inorderTraversal(root.right)

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

注:

1. 类中方法的自我调用

2. Python中list可以直接相加得到新的list:

ls1 = [1,2,3]

ls2 = [4,5,6]

print(ls1+ls2)

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

迭代:

class Solution:

def inorderTraversal(self, root: TreeNode) -> List[int]:

# 迭代解法

p = root

res = []

stack = []

while p or stack:

if p:

stack.append(p)

p = p.left

else:

tmp = stack.pop()

res.append(tmp.val);

p = tmp.right

return res

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

(二)二叉树的先序(前序)遍历

递归:

class Solution:

def preorderTraversal(self, root: TreeNode) -> List[int]:

‘‘‘递归解法‘‘‘

p =root

res = []

if p!=None:

res.append(p.val)

res += self.preorderTraversal(p.left)

res += self.preorderTraversal(p.right)

return res

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

迭代:

class Solution:

def preorderTraversal(self, root: TreeNode) -> List[int]:

‘‘‘迭代解法‘‘‘

p = root

res = []

stack = []

while p or stack:

if p:

res.append(p.val)

stack.append(p)

p = p.left

else:

temp = stack.pop()

p = temp.right

return res

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

(三)二叉树的后序遍历

递归:

class Solution:

def postorderTraversal(self, root: TreeNode) -> List[int]:

p = root

res = []

if p:

res += self.postorderTraversal(p.left)

res += self.postorderTraversal(p.right)

res.append(p.val)

return res

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

已有详细解释说明,不再说明。

迭代1:

class Solution:

def postorderTraversal(self, root: TreeNode) -> List[int]:

‘‘‘先序遍历思想实现后续遍历‘‘‘

p = root

#res = []

stack = []

stack2 = []

while p or stack:

if p:

stack2.append(p.val)

stack.append(p)

p = p.right

else:

temp = stack.pop()

p = temp.left

return stack2[::-1]

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

迭代2:

class Solution:

def postorderTraversal(self, root: TreeNode) -> List[int]:

‘‘‘后序遍历双指针迭代算法‘‘‘

if not root: # 需要判断是否为空

return []

stack = []

res = []

prev = None

curr = None

stack.append(root)

while stack:

curr = stack[-1]

if prev==None or prev.left==curr or prev.right==curr:

if curr.left!=None:

stack.append(curr.left)

elif curr.right!=None:

stack.append(curr.right)

elif prev == curr.left:

if curr.right!=None:

stack.append(curr.right)

else:

res.append(curr.val)

stack.pop() # 需要弹出

prev = curr

return res

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

(四)二叉树的层次遍历

采用队列组织结构

class Solution:

def levelOrder(self, root: TreeNode) -> List[List[int]]:

if not root:

return []

queue = []

res = []

p = root

queue.append(p)

while queue:

temp = queue.pop(0)

res.append(temp.val)

if temp.left!=None:

queue.append(temp.left)

if temp.right!=None:

queue.append(temp.right)

return res

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

原文:https://www.cnblogs.com/ACStrive/p/11222390.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值