【剑指Offer】面试题39：二叉树的深度

一：题目描述

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

二：解题思路

1.根节点为空，树的深度为0

2.只有一个根节点，树的深度为1

3.有左子树或右子树或两者均有，树的深度为 1+左右子树深度的最大值

三：代码实现

/*
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)
    {
    
        //根节点为空，深度0
        if(pRoot==NULL)
            return 0;
        //只包含根节点，深度1
        if(pRoot->left==NULL && pRoot->right==NULL)
            return 1;
        
        int depthLeftTree=1;
        int depthRightTree=1;
        //树有左子树或右子树，或两者都有，返回两边的最大值
        if(pRoot->left)
           depthLeftTree+=TreeDepth(pRoot->left);
        if(pRoot->right)
            depthRightTree+=TreeDepth(pRoot->right);
        
        return depthLeftTree>depthRightTree?depthLeftTree:depthRightTree;
    }
};