题解 | #JZ55 二叉树的深度#
用递归的写法，把它分成独立的子任务来看待，只考虑子任务和当前结点之间的关系。子任务都当作是已经完成了的。
子任务具体的完成实现是在函数出口中实现。
这里我们只考虑当前结点和左子树的最大深度以及和右子树最大深度之间的关系。
显然关系为当前结点的最大深度=max(左子树最大深度, 右子树最大深度) + 1
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
int TreeDepth(TreeNode* pRoot) {
if (!pRoot) return 0;
if (!pRoot->left && !pRoot->right) return 1;
return max(TreeDepth(pRoot->left), TreeDepth(pRoot->right)) + 1;
}
};
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
#
# @param pRoot TreeNode类
# @return int整型
#
class Solution:
def TreeDepth(self , pRoot: TreeNode) -> int:
# write code here
if pRoot is None:
return 0
if pRoot.left is None and pRoot.right is None:
return 1
return max(self.TreeDepth(pRoot.left), self.TreeDepth(pRoot.right)) + 1
CVTE公司福利 270人发布
