题目
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
思路
这道题蛮简单的,求二叉树的深度。可以是递归的方法,属于DFS(深度优先搜索);另一种方法是按照层次遍历,属于BFS(广度优先搜索)。
代码:
DFS:
struct TreeNode
{
int val;
TreeNode *left;
TreeNode *right;
}
class Solution
{
public:
int TreeDepth(TreeNode* pRoot)
{
if(pRoot == NULL)
{
return 0;
}
int nleft = TreeDepth(pRoot->left);
int nright = TreeDepth(pRoot->right);
return (left>right)?(left+1):(right+1);
}
};
BFS:
class Solution
{
public:
int TreeDepth(TreeNode* pRoot)
{
if(pRoot == NULL)
return 0;
queue<TreeNode*>que;
int depth=0;
que.push(pRoot);
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);
}
}
}
};