剑指offer 39.判断二叉树是否为平衡二叉树

题目

题目描述

输入一棵二叉树，判断该二叉树是否是平衡二叉树。

思路：如果直接使用递归遍历，会重复遍历之前的节点，更好的方法是采用后序遍历，对每个节点来说，我们都已经遍历了它的左右子树，所以边遍历边判断，最后遍历到树的根节点时，判断完毕。

递归遍历：

class Solution {
public:
    
    bool IsBalanced_Solution(TreeNode* pRoot) {
		if(pRoot==NULL) return true;
        int left=IsBalanced_Solution(pRoot->left);
        int right=IsBalanced_Solution(pRoot->right);
      	int diff=left-right;
        if(diff>1||diff<-1)
            return false;
       return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
    }
};

后序遍历：

class Solution {
public:
    bool IsBalanced(TreeNode* pRoot,int* pDepth){
        if(pRoot==NULL){
            *pDepth=0;
            return true;
        }
        int left,right;
        if(IsBalanced(pRoot->left,&left)&&IsBalanced(pRoot->right,&right)){
	      	int diff=left-right;      
            if(diff>=-1&&diff<=1){
            *pDepth=(left>right)?(left+1):(right+1);
            return true;
            }
        }
        return false;
    }
    
    bool IsBalanced_Solution(TreeNode* pRoot) {
		int Depth=0;
        return IsBalanced(pRoot,&Depth);
    }
};