题目描述:输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
解题思路:指针指向队头元素然后队头元素出队;队头元素如果有左右孩子依次入队。
迭代代码:
class Solution {
public:
int maxDepth(TreeNode* root) {
queue<TreeNode*>q;
if(root) q.push(root);
int depth=0;
while(q.size()){
int n=q.size();
for(int i=0;i<n;i++){
TreeNode*tr=q.front();
q.pop();
if(tr->left) q.push(tr->left);
if(tr->right) q.push(tr->right);
}
depth++;
}
return depth;
}
};
递归代码:
class Solution {
public:
int maxDepth(TreeNode* root) {
return !root ? 0 : max(maxDepth(root->left),maxDepth(root->right)) + 1;
}
};