剑指offer 面试题55 二叉树的深度

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

思路：递归

树的深度为左子树的深度和右子树的深度的较大值+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)
    {
        if(!pRoot)
            return 0;
        int left=TreeDepth(pRoot->left);
        int right=TreeDepth(pRoot->right);
        return left>right?left+1:right+1;
    }
};

复杂度分析：时间复杂度为O(n)，空间复杂度为O(1).

问题2：输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

思路：后序遍历。遍历每个节点时，一边记录节点的深度，一边判断是否为平衡二叉树。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isBalancedCore(TreeNode* node, int& depth)
    {
        if(node==nullptr)
        {
            depth=0;
            return true;
        }
        int left,right;
        if(isBalancedCore(node->left,left)&&isBalancedCore(node->right,right))
        {
            int diff=left-right;
            if(diff<=1&&diff>=-1)
            {
                depth=1+(left>right?left:right);
                return true;
            }
        }
        return false;
    }
    bool isBalanced(TreeNode* root) {
        int depth=0;
        return isBalancedCore(root,depth);
    }
};

复杂度分析：时间复杂度为O(n)，空间复杂度为O(n)。