求二叉樹的深度

思路：

分別遞迴左右子樹，深度=左右子樹中大的一個+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==NULL)
            {
            return 0;
        }
        int leftDepth=TreeDepth(pRoot->left);
        int rightDepth=TreeDepth(pRoot->right);
        return leftDepth>rightDepth?leftDepth+1:rightDepth+1;
    }
};
 

輸入一顆二叉樹的根節點，判斷該樹是不是平衡二叉樹

思路：

遞迴每一層都分別算出左右子樹的高度，比較，看是否平衡

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        //判斷平衡，比較高度
        if(pRoot==NULL)
            {
            return true;   //空樹是平衡的二叉樹
        }
        int leftDep=TreeDepth(pRoot->left);
        int rightDep=TreeDepth(pRoot->right);
        int dif=leftDep-rightDep;
        if(dif>1||dif<-1)
            {
            return false;
        }
        return IsBalanced_Solution(pRoot->left)&&IsBalanced_Solution(pRoot->right);
    }
    
    int TreeDepth(TreeNode* pRoot)
        {
        if(pRoot==NULL)
            {
            return 0;
        }
        int leftDepth=TreeDepth(pRoot->left);
        int rightDepth=TreeDepth(pRoot->right);
        return leftDepth>rightDepth?leftDepth+1:rightDepth+1;
    }
};