剑指offer系列31：二叉树的深度

第一个小题目是求二叉树的深度的，前面做过很多二叉树的题，这个就比较简单了。

class Solution {
public:
    int max(int x, int y)
    {
        return (x > y) ? x : y;
    }
    int TreeDepth(TreeNode* pRoot)
    {
        if (pRoot == NULL)
            return 0;
        int left = 0, right = 0,depth=0;
        if (pRoot->left)
        {
            left = TreeDepth(pRoot->left);
        }
        if (pRoot->right)
        {
            right = TreeDepth(pRoot->right);
        }
        return max(left,right)+1;

    }
};

第二个小题在第一个题的基础上也不难，主要是对于递归的掌握。

class Solution {
public:
    
    int max(int x, int y)
    {
        return (x > y) ? x : y;
    }
    
    int TreeDepth(TreeNode* pRoot)
    {
        if (pRoot == NULL)
            return 0;

        int left = 0, right = 0;
        left = TreeDepth(pRoot->left);//这里不用判断左子树和右子树是否为空，因为如果是空也能进入函数，而且前面两行可以终止递归
        right = TreeDepth(pRoot->right);
        return max(left,right)+1;

    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if (pRoot == NULL)
            return true;
        //分别计算左右子树的深度
        int left = TreeDepth(pRoot->left);
        int right = TreeDepth(pRoot->right);
        if (abs(left - right) <= 1)
            return IsBalanced_Solution(pRoot->left)&& IsBalanced_Solution(pRoot->right);
        else
             return false;
    }
};

对于树的题目，脑子里不要想太复杂的东西，按照以下步骤做就好：

1.写递归终结条件，一般是结点指针为空

2.把树想象成最简单只有3个结点的小树，想想该怎么计算

3.写递归条件，让它可以应对更复杂的树

看到书上还有一种做法，因为之前要重复的遍历结点计算是否为平衡二叉树。所以用后序遍历的方式遍历每一个结点，在遍历结点之前已经遍历了它的左右子树。前一种方法用的是前序遍历。

class Solution {
public:
    
    bool IsBalanced_Solution(TreeNode* pRoot,int *depth) {
        if (pRoot == NULL)
        {
            *depth = 0;
            return true;
        }
        int left=0, right=0;//分别为左右子树的深度
        if (IsBalanced_Solution(pRoot->left,&left) && IsBalanced_Solution(pRoot->right,&right))
        {
            int dif = left - right;
            if (dif <=1&&dif>=-1)
            {
                *depth = 1+(left > right ? left : right);//在返回是否为平衡二叉树之前计算深度
                return true;
            }
        }
        return false;
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        int depth = 0;
        return IsBalanced_Solution(pRoot, &depth);
    }
};