「代码随想录」Python刷题笔记Day14-二叉树1

六个噜酱

已于 2023-07-29 23:46:14 修改

阅读量134

点赞数

分类专栏：代码随想录文章标签： python 笔记 leetcode

于 2023-06-09 20:00:14 首次发布

本文链接：https://blog.csdn.net/weixin_44622140/article/details/131124745

版权

代码随想录专栏收录该内容

13 篇文章 3 订阅

订阅专栏

二叉树第一天

小白一刷代码随想录（Python），主打跟上进度和学习。二刷时会回来更新博文。
本文属于个人笔记，如有问题欢迎指出。

二叉树的理论基础

卡哥链接

二叉树的递归遍历

卡哥链接

前序遍历
leetcode-144

class Solution:
    def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        if not root:
            return []
        
        left = self.inorderTraversal(root.left)
        right = self.inorderTraversal(root.right)
        return [root.val] + left + right

中序遍历
leetcode-

class Solution:
    def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        if not root:
            return []
        
        left = self.inorderTraversal(root.left)
        right = self.inorderTraversal(root.right)
        return left + [root.val] + right

后序遍历
leetcode-145

class Solution:
    def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        if not root:
            return []
        
        left = self.inorderTraversal(root.left)
        right = self.inorderTraversal(root.right)
        return left + right + [root.val]

举个简单例子: 合并二叉树

合并的规则是：如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；否则，不为 null 的节点将直接作为新二叉树的节点。返回合并后的二叉树。
在这里插入图片描述

# 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 mergeTrees(self, root1: Optional[TreeNode], root2: Optional[TreeNode]) -> Optional[TreeNode]:
        if not root1:
            return root2
        if not root2:
            return root1

        head = TreeNode(root1.val + root2.val)
        head.left = self.mergeTrees(root1.left, root2.left)
        head.right = self.mergeTrees(root1.right, root2.right)
        return head

二叉树的迭代

卡哥链接

前序遍历
leetcode-144

class Solution:
    def preorder(self, res, root: TreeNode):
        if not root:
            return
        res.append(root.val)
        self.preorder(res, root.left)
        self.preorder(res, root.right)

    def preorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        res = []
        self.preorder(res, root)
        return res

中序遍历
leetcode-94

class Solution:
    def inorder(self, res, root):
        if not root:
            return []
        self.inorder(res, root.left)
        res.append(root.val)
        self.inorder(res, root.right)
        return res

    def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        res = []
        res = self.inorder(res, root)
        return res

后序遍历
leetcode-145

class Solution:
    def inorder(self, res, root):
        if not root:
            return []
        self.inorder(res, root.left)
        self.inorder(res, root.right)
        res.append(root.val)
        return res

    def postorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        res = []
        res = self.inorder(res, root)
        return res