剑指Offer_编程题38:二叉树的深度

题目:输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。

牛客网:链接

思路:如果一棵树只有一个节点,那么它的深度是1。如果根节点只有左子树而没有右子树,那么树的深度应该是其左子树的深度加1,;同样,如果根节点只有右子树而没有左子树,那么树的深度应该是其右子树的深度加1。如果既有右子树又有左子树,那么树的深度就是其左、右子树深度的较大值加1。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def TreeDepth(self, pRoot):
        # write code here
        if not pRoot:
            return 0
        if not pRoot.left and not pRoot.right:
            return 1
        return max(self.TreeDepth(pRoot.left), self.TreeDepth(pRoot.right)) + 1

猜你喜欢

转载自blog.csdn.net/mengmengdajuanjuan/article/details/82899195