输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
1
2
3
4
5
6
7
|
class
Solution {
public
:
int
TreeDepth(TreeNode* pRoot){
if
(!pRoot)
return
0 ;
return
max(1+TreeDepth(pRoot->left), 1+TreeDepth(pRoot->right));
}
};
|
int TreeDepth(TreeNode* pRoot)
{
queue<TreeNode*> q;
if(!pRoot) return 0;
q.push(pRoot);
int level=0;
while(!q.empty()){
int len=q.size();
level++;
while(len--){
TreeNode* tem=q.front();
q.pop();
if(tem->left) q.push(tem->left);
if(tem->right) q.push(tem->right);
}
}
return level;
}
非递归写法:层次遍历
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
import
java.util.Queue;
import
java.util.LinkedList;
public
class
Solution {
public
int
TreeDepth(TreeNode pRoot)
{
if
(pRoot ==
null
){
return
0
;
}
Queue<TreeNode> queue =
new
LinkedList<TreeNode>();
queue.add(pRoot);
int
depth =
0
, count =
0
, nextCount =
1
;
while
(queue.size()!=
0
){
TreeNode top = queue.poll();
count++;
if
(top.left !=
null
){
queue.add(top.left);
}
if
(top.right !=
null
){
queue.add(top.right);
}
if
(count == nextCount){
nextCount = queue.size();
count =
0
;
depth++;
}
}
return
depth;
}
}
|
递归写法,比较简单,不解释:
1
2
3
4
5
6
7
8
9
10
11
12
|
import
java.lang.Math;
public
class
Solution {
public
int
TreeDepth(TreeNode pRoot)
{
if
(pRoot ==
null
){
return
0
;
}
int
left = TreeDepth(pRoot.left);
int
right = TreeDepth(pRoot.right);
return
Math.max(left, right) +
1
;
}
}
|