题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
解:对于二叉树深度的求解可以采用递归和迭代两种思路解题。
递归比较简单。
int TreeDepth(TreeNode*pRoot)
{
if(!pRoot)
return 0;
int left=TreeDepth(pRoot->left);
int right=TreeDepth(pRoot->right);
return left>=right?(left+1):(right+1);
}
迭代法:
经典的非递归层次遍历:利用辅助队列,先将头节点入队列,当队列不空时出队列的节点记为current,
当current左节点不空时入队列,其右节点不空时入队列,如此循环即可。
求深度:构造变量cur记录当前层访问到的节点数,width记录当前层的总个数,每当访问过一层层数deep++;
此种方法同时可以求最大宽度,访问第几层的第几个节点,是一种通用方法!
/*
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)
return 0;
queue<TreeNode*> myQueue;
myQueue.push(pRoot);
int depth=0;
while(!myQueue.empty())
{
depth++;
for(int i=0;i<myQueue.size();i++)
{
TreeNode* temp=myQueue.front();
myQueue.pop();
if(temp->left) myQueue.push(temp->left);
if(temp->right) myQueue.push(temp->right);
}
}
return depth;
}
};