题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
思路:
求树的深度,可以从层次遍历出发考虑。
层次遍历可以使用队列完成,也可以使用递归完成,所以有两种方法
--------------------------------------------------------------------------------------------------------
递归分析一下:
思路: 从跟节点出发, 查询左子树的深度 , 获取右子树的深度,比较一下,取大的,再加一 。就是整个二叉树的深度
递归的三个条件
边界条件:当前节点下,是否还有子节点,没有返回,有继续递归
递归前进段:当前节点下,还有子节点;
递归返回段:当前节点下,没有子节点。
// 递归遍历,仅仅一行
class Solution {
public:
int TreeDepth(TreeNode* pRoot){
return pRoot ? max(TreeDepth(pRoot->left),TreeDepth(pRoot->right))+1 :0;
}
};
//迭代版本
class Solution {
public:
int TreeDepth(TreeNode* pRoot) {
if (!pRoot) return 0;
queue<TreeNode*> que;
que.push(pRoot);int depth=0;
while (!que.empty()) {
int size=que.size();
depth++;
for (int i=0;i<size;i++) { //一次处理一层的数据
TreeNode *node=que.front();
que.pop();
if (node->left) que.push(node->left);
if (node->right) que.push(node->right);
}
}
return depth;
}
};
介绍下使用层次遍历计算二叉树的深度的思路:
借助队列,对二叉树进行层次遍历;
- 在层次遍历的过程中,每次当队列中某一层的节点出队完成后,高度+1;
- 关键点:判别队列中某一层节点出队完成的标准是什么?
在出队之前,此时队列中记录的只有某一层节点,所以队列的大小就是某一层节点的个数。当此个数减到0的时候,则说明该层节点全部出队完成 -
public int TreeDepth2(TreeNode root) { if(root == null) return 0; Queue<TreeNode> queue = new LinkedList(); queue.add(root); int high = 0; int size;//用size控制high增长的次数和时机(同一层的元素没有完全退出队列的时候high不可以增加) TreeNode node; while(queue.size() != 0){ size = queue.size();//队列长度 while(size != 0){ node = queue.poll(); if(node.left != null) queue.add(node.left); if(node.right != null) queue.add(node.right); size--;//当size==0时说明同一层的元素遍历完成 } high++; } return high; }