题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
递归解题思路:
1、如果是空树,返回0;
2、如果只有根结点,返回1;
3、否则递归遍历左右子树;
4、返回左右子树较大者+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)
{
/*
解题思路:
1、如果是空树,返回0;
2、如果只有根结点,返回1;
3、否则递归遍历左右子树;
4、返回左右子树较大者+1(根结点).
*/
if(pRoot==NULL){
return 0;
}
if(pRoot->right==NULL && pRoot->left == NULL){
return 1;
}
int leftLen,rightLen;
leftLen = TreeDepth(pRoot->left);
rightLen = TreeDepth(pRoot->right);
int h = leftLen>rightLen?leftLen:rightLen;
return h+1;
}
};
效率:
非递归解题思路:
1、根结点入队列,记录当前结点数;
2、如果队列非空,队首元素出队,结点数加1;
3、左孩子非空,左孩子入队,如果右孩子非空,右孩子入队;
4、遍历完所有当前层的结点,深度加1,结点数清零,重新开始计数;
5、遍历完整棵树,队列为空,返回深度即为树的深度
代码实现:
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
import java.util.LinkedList;
import java.util.Queue;
public class Solution {
public int TreeDepth(TreeNode root) {
/*
解题思路:
1、根结点入队列,记录当前结点数;
2、如果队列非空,队首元素出队,结点数加1;
3、左孩子非空,左孩子入队,如果右孩子非空,右孩子入队;
4、遍历完所有当前层的结点,深度加1,结点数清零,重新开始计数;
5、遍历完整棵树,队列为空,返回深度即为树的深度
*/
if(root==null){
return 0;
}
Queue<TreeNode> q = new LinkedList<TreeNode>();
int depth = 0;
int count = 0,nextCount;
// 1、根结点入队列,记录当前结点数;
q.offer(root);
nextCount = q.size();
while(q.size()!=0){
// 2、如果队列非空,队首元素出队,结点数加1;
TreeNode currentNode = q.poll();
count++;
// 3、左孩子非空,左孩子入队,如果右孩子非空,右孩子入队;
if(currentNode.left!=null){
q.offer(currentNode.left);
}
if(currentNode.right!=null){
q.offer(currentNode.right);
}
//4、遍历完所有当前层的结点,深度加1,结点数清零,重新开始计数
if(count == nextCount){
depth++;
count = 0;
nextCount = q.size();
}
}
return depth;
}
}