剑指offer-二叉树的深度

 

题目描述

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

 

解题思路

max_dep保存最大深度，cur_dep保存当前遍历节点的深度，均初始化为0。利用前序遍历的思想，每次访问到一个节点，先对cur_dep加1，若当前节点为叶子结点，那么比较cur_dep与max_dep，若cur_dep更大则更新max_dep为当前节点深度。遍历完当前节点的左右子节点之后，对cur_dep减1，回溯到父节点继续遍历右子节点。当遍历完整棵树时，max_dep中保存的便是最大深度。

代码

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {
public:
    int max_dep = 0;
    int cur_dep = 0;
    int TreeDepth(TreeNode* pRoot)
    {
        if(pRoot != NULL){
            cur_dep++;
            if(pRoot->left)
                TreeDepth(pRoot->left);
            if(pRoot->right)
                TreeDepth(pRoot->right);
            if(pRoot->left == NULL&&pRoot->right == NULL)
                if(cur_dep > max_dep)
                    max_dep = cur_dep;
            cur_dep--;
            return max_dep;
        }
        return max_dep;
    }
};