求二叉树的深度 python

Posted 2022-10-15 fengff

 题目描述

输入一棵二叉树，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。
解题思路

两种解法，一种是第一时间的思路，即获得根节点到每个叶节点的深度，取其中最长的返回：

class Solution:
    def __init__(self):
        self.maxDeep = 0
        self.curDeep = 0

    def TreeDepth(self, pRoot):
        if not pRoot:
            return 0
        self.curDeep += 1
        self.TreeDepth(pRoot.left)
        self.maxDeep = self.curDeep if self.curDeep > self.maxDeep else self.maxDeep
        self.TreeDepth(pRoot.right)
        self.maxDeep = self.curDeep if self.curDeep > self.maxDeep else self.maxDeep
        self.curDeep -= 1


这种方案可以解决问题，但是代码不好看，还定义了两个公共变量。第二种解法是自底而上，每当从子节点回退到其父节点时，返回左右子节点中最深的分支值，没有公共变量，代码短小精悍：

def TreeDeep(self, pRoot):
    if not pRoot:
        return 0
    left = self.TreeDeep(pRoot.left)+1
    right = self.TreeDeep(pRoot.right)+1
    return left if left>right else right



如果想进一步缩短代码高度，那就把left和right的表达式带入第6行代码中。虽然代码高度进一步缩短，但是增加了代码宽度，且降低了代码的可读性，《Clean Code》中所不建议的方式，所以就到此为止了~
————————————————
版权声明：本文为CSDN博主「goddaniel」的原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/u010005281/article/details/79659747

　　二叉树有深度和高度两个属性，一个节点的深度指的是从根节点到该节点路径的长度，根节点的深度为1；一个节点的高度指的是从该节点到叶子节点所有路径上包含节点个数的最大值。叶子节点的高度为1，往上节点的高度依次递增。所以要求二叉树的深度，我们要求出从根节点到叶子结点最长路径的长度，从根节点到所有的叶子结点，实际就是在遍历这棵树，使用深度优先遍历可以解决这个问题。

　　首先还是定义一颗二叉树的类。

1 class TreeNode:
2     def __init__(self, x):
3         self.val = x
4         self.left = None
5         self.right = None

　　然后，为了得到从根节点到叶子结点路径的长度，我们从根节点出发，进行深度优先遍历，当遍历到一个叶子结点时，就计算出了一条路径的长度，这时返回到上一层，即叶子结点的父亲节点，继续对另一颗子树进行深度优先遍历，计算其他路径的长度。因此，这里有两个问题，第一，我们需要一个变量保存当前遍历路径的长度，它保存的一定是当前遍历的路径长度，应该作为函数的参数（深度优先遍历的函数）；第二，当遍历到叶子节点时，我们要判断这条路径的长度是否是最长的，因此还需要一个变量来保存当前最长的路径长度，它不会随着函数的迭代改变，应当是一个全局变量，代码如下。

class Solution:
     #二叉树的深度
    def TreeDepth(self, pRoot):
    # write code here
        global maxdepth#全局变量保存当前最长的路径长度
        maxdepth=0
        def DfsTree(root,depth=0):#depth是当前的路径长度
            global maxdepth
            if root:#节点非空，路径长度加1
                depth+=1
            if root.left:#左子树非空，继续遍历左子树
                DfsTree(root.left,depth)
            if root.right:#右子树非空，继续遍历右子树
                DfsTree(root.right,depth)
            if not root.left or root.right:#左右子树都是空的，即该节点是一个叶子结点，则判断depth是不是最长的路径
                if depth>maxdepth:
                    maxdepth=depth
        if root:
            DfsTree(root)
        return maxdepth