题目描述
输入一棵二叉树的根结点,求该树的深度。
从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
数据范围
树中节点数量 [0,500]。
样例
输入:二叉树[8, 12, 2, null, null, 6, 4, null, null, null, null]如下图所示: 8 / \ 12 2 / \ 6 4 输出:3
方法一:dfs O(n)
我们可以递归计算深度,先递归到树的叶子结点,然后开始回溯,每次回溯返回的是当前结点的深度,公式如下:
m a x ( 左子树深度 , 右子树深度 ) + 1 max(左子树深度,右子树深度)+1max(左子树深度,右子树深度)+1
举个例子,那题目样例来看,得到的结果如下图所示:
/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */ class Solution { public: int treeDepth(TreeNode* root) { if (root == NULL) return 0; int left = treeDepth(root->left); int right = treeDepth(root->right); return max(left, right) + 1; } };
欢迎大家在评论区交流~