前序遍历解法
前序遍历求二叉树深度
class Solution {
public:
int res = 0;
void dfs(TreeNode* root, int depth) {
res = res > depth ? res : depth;
if (root->left == nullptr && root->right == nullptr) return;
if (root->left) {
depth ++;
dfs(root->left, depth);
depth --;
}
if (root->right) {
depth ++;
dfs(root->right, depth);
depth --;
}
}
int maxDepth(TreeNode* root) {
if (root == nullptr) return res;
dfs(root, 1);
return res;
}
};
后序遍历解法
后序遍历求二叉树高度
class Solution {
public:
int dfs(TreeNode* node) {
if (node == nullptr) return 0;
int left = dfs(node->left);
int right = dfs(node->right);
return 1 + max(left, right);
}
int maxDepth(TreeNode* root) {
return dfs(root);
}
};