常见的二叉树的类型:
满二叉树:每一层节点都是满的
完全二叉树:除了最后一层,其余所有层的节点都是满的。并且最后一层的节点必须从左到右连续的(从上到下,从左到右,节点的编号必须连续)
二叉搜索树:左节点必须小于当前节点的值。右节点必须大于当前结点的值
平衡二叉搜索树:左子树和右子树高度之差绝对值不大于一的二叉搜索树。
遍历方式:递归遍历/迭代遍历。递归遍历又分前序、中序、层序三种,一刷先掌握递归遍历。
具体题目:
LC 144 二叉树的前序遍历
class Solution:
def preorder(self,root,res=[]):
if root:
res.append(root.val)
self.preorder(root.left , res)
self.preorder(root.right , res)
return res
def preorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
res = []
self.preorder(root , res)
return res
中序:中左右
后序:左右中
使用队列模拟层序遍历,即逐层从上到下,从左到右遍历树的元素。
class Solution:
def levelOrder(self, root: Optional[TreeNode]) -> List[List[int]]:
res , queue = [] , []
if root == None:
return []
queue.append(root)
while queue:
temp = []
for i in range(len(queue)):
node = queue.pop(0)
temp.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
res.append(temp)
return res
首先初始化结果集、队列都为[]。(可以使用python中的标准库collections,其中的dequeue可以将pop(0)的时间复杂度从O(N)降低至O(1))
把根节点加到队列中,很自然此时队列长为1,再把根节点弹出并把节点值加入到temp中,temp是保存每一层元素的结果缓存列表。
若这个树只有根节点,那么弹出根节点后,队列长为0,那么可直接return掉。
若根节点还有左右儿子,那么将其左右儿子加入到队列中,此时队列长为2 。再依次弹出两个根节点并把值加入到temp中,,,依次类推
当某一次for循环遍历队列后,队列为空,说明树已经被遍历完了,没有下一层了。return。
关键要理解:for循环遍历第k层节点的同时,也在将k+1层的节点添加到队列尾,for循环的次数就是第k层节点的个数,当for循环结束,第k层的节点全被弹出,此时队列中全是k+1层的节点。
相当于层序遍历+反转列表,算是个复习
class Solution:
def levelOrderBottom(self, root: Optional[TreeNode]) -> List[List[int]]:
res , queue = [] , collections.deque()
if not root:
return []
queue.append(root)
while queue:
temp = []
for _ in range(len(queue)):
node = queue.popleft()
temp.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
res.append(temp)
left , right = 0 , len(res) -1
while left <= right:
res[left] , res[right] = res[right] , res[left]
left += 1
right -= 1
return res
#层序遍历,每次temp数组添加的是每一层树的最后一个节点值
class Solution:
def rightSideView(self, root: Optional[TreeNode]) -> List[int]:
res , queue = [] , collections.deque()
if not root:
return []
queue.append(root)
while queue:
temp = []
for i in range(len(queue)):
node = queue.popleft()#先弹出队列中最左边的节点,再取值
temp.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
res.append(temp[-1])
return res