判断是否为平衡二叉树--递归法

求树的深度

用递归做很简单，只要知道递归出口语句的别写错。

[cpp]  view plain copy

1. struct BinaryTreeNode  
2. {  
3.     int m_Value;  
4.     BinaryTreeNode* m_pLeft;  
5.     BinaryTreeNode* m_pRight;  
6. };  
7.   
8. int TreeDepth(BinaryTreeNode* pRoot)  
9. {  
10.     if (pRoot == NULL)  
11.         return 0;  
12.   
13.     int nLeftDepth = TreeDepth(pRoot->m_pLeft);  
14.     int nRightDepth = TreeDepth(pRoot->m_pRight);  
15.   
16.     return (nLeftDepth>nRightDepth)?(nLeftDepth+1):(nRightDepth+1);  
17. }  

判断该树是否为平衡二叉树

调用上述函数求每个节点的左右孩子深度

[cpp]  view plain copy

1. bool IsBalanced(BinaryTreeNode* pRoot)  
2. {  
3.     if(pRoot== NULL)  
4.         return true;  
5.   
6.     int nLeftDepth = TreeDepth(pRoot->m_pLeft);  
7.     int nRightDepth = TreeDepth(pRoot->m_pRight);  
8.     int diff = nRightDepth-nLeftDepth;  
9.   
10.     if (diff>1 || diff<-1)  
11.         return false;  
12.   
13.     return IsBalanced(pRoot->m_pLeft)&&IsBalanced(pRoot->m_pRight);  
14. }