剑指offer之二叉树的深度

1.题目描述

输入一棵二叉树，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。

2.问题分析

方法一：

可以使用递归的方法，分别求左子树和右子树的深度，则树的深度是：更大者 + 1。这是一个递归的方法。

方法二：

使用层次遍历的方法，每访问一层，则深度+1，所有层的访问结束之后，就等到深度了。这里需要使用到一个队列，来帮助我们层次遍历，这不是一个递归的方法

3.源代码

方法一：

int TreeDepth(TreeNode* pRoot)
{
    if(pRoot == NULL)
        return 0;
    int left = TreeDepth(pRoot->left);
    int right = TreeDepth(pRoot->right);
    return (left + 1 > right + 1) ? left + 1:right + 1;
}

方法二：

int TreeDepth(TreeNode* pRoot)
{
    if(pRoot == NULL)
        return 0;
    queue<TreeNode*> nodes;
    nodes.push(pRoot);
    int deep = 0;
    int curCount = 1;
    int nextCount = 0;
    while(!nodes.empty())
    {
        TreeNode* node = nodes.front();
        nodes.pop();
        --curCount;
        if(node->left != NULL)
        {
            ++nextCount;
            nodes.push(node->left);
        }
        if(node->right != NULL)
        {
            ++nextCount;
            nodes.push(node->right);
        }
        if(curCount == 0)
        {
            curCount = nextCount;
            nextCount = 0;
            ++deep;
        }
    }
    return deep;
}