【劍指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;
    }
};