题目来源:二叉树的深度_牛客题霸_牛客网 (nowcoder.com)
描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度,根节点的深度视为 1 。
数据范围:节点的数量满足 0 \le n \le 1000≤n≤100 ,节点上的值满足 0 \le val \le 1000≤val≤100
进阶:空间复杂度 O(1)O(1) ,时间复杂度 O(n)O(n)
假如输入的用例为{1,2,3,4,5,#,6,#,#,7},那么如下图:
我的答案:
我直接用了递归的方法,简单粗暴,和答案的第一种方法一样(毫不意外)。
/*
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==nullptr) return 0;
int valL=TreeDepth(pRoot->left);
int valR=TreeDepth(pRoot->right);
return max(valL,valR)+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==nullptr) return 0;
queue<TreeNode*> q;
int level=0;
q.push(pRoot);
while(!q.empty()){
int si=q.size();
while(si--){
auto node=q.front();
q.pop();
if(node->left) q.push(node->left);
if(node->right) q.push(node->right);
}
level++;
}
return level;
}
};