LC-104.二叉树的最大深度

LC-104.二叉树的最大深度

这道题是一道简单题,可以使用递归法和迭代法进行求解。

迭代法

迭代法最好就使用队列进行层序遍历,这样每遍历一层,深度+1:

class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        if not root: return 0
        depth = 0
        
        from collections import deque
        que = deque([root])
        while que:
            depth += 1
            size = len(que)
            for _ in range(size):
                cur = que.popleft()
                if cur.left: que.append(cur.left)
                if cur.right: que.append(cur.right)
        return depth

递归法

递归法这部分其实可以分为前序遍历(中左右),也可以使用后序遍历(左右中)。使用前序求的就是深度,使用后序求的是高度,而根节点的高度就是二叉树的最大深度

  • 前序(中左右)

前序其实就是在递归之前就已经将本层的深度加进去,然后再进行递归,是从上至下的。这体现出了从上之下逐层递归+1的思想。

class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        depth = 0
        def digui(root, depth):
        	# 这里直接返回了depth,所以上一层递归的时候,不需要额外判断左右孩子是否存在。
            if not root: return depth

            l_d = digui(root.left, depth + 1)
            r_d = digui(root.right, depth + 1)
            return max(l_d, r_d)

        return digui(root, depth)
  • 后序遍历(左右中)

后序遍历就是每次从下至上的。也就说到最底层后,每次向上一层返回时,再取左右子树最高的高度后+1,这体现出了从下至上的求高度的过程。

class solution:
    def maxdepth(self, root: treenode) -> int:
        return self.getdepth(root)
        
    def getdepth(self, node):
        if not node:
            return 0
        leftdepth = self.getdepth(node.left) #左
        rightdepth = self.getdepth(node.right) #右
        depth = 1 + max(leftdepth, rightdepth) #中
        return depth

559. N 叉树的最大深度

这里有一道类似的题,思想是一样的,只不过递归的时候,因为是N叉树,不是直接递归左右子树,而是用一个循环依次递归children。每次递归用一个tmp记录,然后找到这个tmp最大值(也就是高度最大的子树),然后加一。

这里有个需要注意的点,如果某个节点没有子树的话,我们循环的时候就会报错,需要提前判断。如果没有children,直接return 1,而不是0。这是为什么呢,因为后序遍历都是先递归返回后再+1,如果没有children没有子树就不会递归,就没有机会去+1,所以我们应该直接去+1.当然,这也就是说,我们的返回条件提前了一层,不是去判断这个节点本身是否存在,而是判断他的children是否存在,如果存在则递归,不存在直接return 1, 而不是0。

class Solution:
    def maxDepth(self, root: 'Node') -> int:
        # 使用后序遍历
        
        def digui(root):
            # if not root:
            #     return 0
            tmp_depth = []
            if not root.children: return 1
            for child in root.children:
                tmp_depth.append(digui(child))
            depth = 1 + max(tmp_depth)
            return depth
        return digui(root)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
以下是用C语言实现的代码: ```c #include <stdio.h> #include <stdlib.h> // 二叉树结点结构体 typedef struct TreeNode { int val; // 结点值 struct TreeNode* left; // 左子结点 struct TreeNode* right; // 右子结点 } TreeNode; // 插入结点 TreeNode* insert(TreeNode* root, int val) { if (root == NULL) { // 如果空树,则插入到根结点 TreeNode* node = (TreeNode*)malloc(sizeof(TreeNode)); node->val = val; node->left = NULL; node->right = NULL; return node; } if (val < root->val) { // 如果插入的值小于当前结点的值,则插入到左子树 root->left = insert(root->left, val); } else { // 否则插入到右子树 root->right = insert(root->right, val); } return root; } // 计算树的高度 int height(TreeNode* root) { if (root == NULL) return 0; int leftHeight = height(root->left); int rightHeight = height(root->right); return (leftHeight > rightHeight ? leftHeight : rightHeight) + 1; } // 打印二叉树 void printTree(TreeNode* root, int layer) { if (root == NULL) return; printTree(root->right, layer + 1); // 先打印右子树 for (int i = 0; i < layer - 1; i++) { printf(" "); } if (layer > 0) { printf(".--"); } printf("%d\n", root->val); printTree(root->left, layer + 1); // 再打印左子树 } int main() { TreeNode* root = NULL; // 根结点 int val; while (scanf("%d", &val) != EOF) { root = insert(root, val); // 插入结点 } int h = height(root); // 计算树的高度 printTree(root, h); // 打印二叉树 return 0; } ``` 输入样例: ``` 10 5 20 8 4 7 ``` 输出样例: ``` .---20 .--10 | `--8 | |--7 `--5 `--4 ``` 解释:该二叉树的结构如下: ``` 10 / \ 5 20 / \ 4 8 / 7 ``` 其中,叶子结点 4 和 7 的高度为 1,结点 5 和 8 的高度为 2,结点 10 的高度为 3,结点 20 的高度为 1。因此,树的高度为 3。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

请叫我哈士奇

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值