94.二叉树的中序遍历
给定一个二叉树的根节点 root
,返回 它的 中序 遍历
方法一 递归
- 前序遍历:打印 - 左 - 右
- 中序遍历:左 - 打印 - 右
- 后序遍历:左 - 右 - 打印
class Solution(object):
def inorderTraversal(self, root):
"""
:type root: TreeNode
:rtype: List[int]
"""
res = []
def dfs(root):
if not root:
return
# 按照 左-打印-右的方式遍历
dfs(root.left)
res.append(root.val)
dfs(root.right)
dfs(root)
return res
方法二 迭代
class Solution(object):
def inorderTraversal(self, root):
"""
:type root: TreeNode
:rtype: List[int]
"""
res = []
stack = []
while stack or root:
# 不断往左子树方向走,每走一次就将当前节点保存到栈中
# 这是模拟递归的调用
if root:
stack.append(root)
root = root.left
# 当前节点为空,说明左边走到头了,从栈中弹出节点并保存
# 然后转向右边节点,继续上面整个过程
else:
tmp = stack.pop()
res.append(tmp.val)
root = tmp.right
return res
方法三 颜色标记法
class Solution:
def inorderTraversal(self, root: TreeNode) -> List[int]:
WHITE, GRAY = 0, 1
res = []
stack = [(WHITE, root)]
while stack:
color, node = stack.pop()
if node is None: continue
if color == WHITE:
stack.append((WHITE, node.right))
stack.append((GRAY, node))
stack.append((WHITE, node.left))
else:
res.append(node.val)
return res
101.对称二叉树
给你一个二叉树的根节点 root
, 检查它是否轴对称。
思路:
543.二叉树的直径
方法一:深度优先搜索
任一路径可看做由某节点为起点,从其左儿子和右儿子向下遍历的路径拼接得到,对应直径:
D = L + R
深度遍历所有节点,设置全局变量ans,返回max(L,ans)
时间复杂度:O(N),其中 N为二叉树的节点数
空间复杂度:O(Height),其中 Height 为二叉树的高度
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:
self.ans = 1
def depth(node):
if not node:
return 0
L = depth(node.left)
R = depth(node.right)
self.ans = max(self.ans, L + R + 1)
return max( L, R) +1
depth(root)
return self.ans -1
102. 二叉树的层序遍历
描述:给你二叉树的根节点 root
,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
思路:广度优先遍历的改进
以队列的方式存储每层的节点
首先根节点进队,当队列不为空时,求当前队列的长度s,依次取出队列所有元素进入下一次迭代,用一个数组存储每层遍历迭代的子节点