【数据结构】中判断一棵树是否为平衡二叉树

平衡二叉树：就是指这棵二叉树的左子树和右子树之间的差值不大于1。

先定义一个二叉树的结构体：

struct TreeNode{TreeNode* _left;TreeNode* _right;};

在判断一颗二叉树是否为平衡二叉树的时候：

第一种方法是，先计算出这棵二叉树的高度。

计算高度的时候使用递归的方法，递归到左子树和右子树。

int TreeDepth(TreeNode* _root)//使用递归方法求出二叉树的高度{if (_root == NULL){return 0;}int left = TreeDepth(_root->_left);int right = TreeDepth(_root->_right);return (left > right) ? (left + 1) : (right + 1);}

再使用判断平衡的函数，计算如果左子树和右子树的差值小于1就是平衡二叉树。

bool IsBalanceTree(TreeNode* _root){if (_root == NULL)//空树也是平衡二叉树{return true;}int left = TreeDepth(_root->_left);//int right = TreeDepth(_root->_right);if ((left - right) > 1 || (left - right) < -1)//平衡二叉树的判断条件，左子树与右子树的高度差不大于1{//如果高度差的绝对值大于1，返回错误return false;}else{return IsBalanceTree(_root->_left) && IsBalanceTree(_root->_right);}}

但是这种方法需要遍历两遍，时间复杂度较高。

下面使用第二种方法，只需要遍历一遍，可以将高度和是否为平衡二叉树判断出来。

bool IsBalanceTree(TreeNode* _root, int &Depth){if (_root == NULL){Depth = 0;return true;}int left, right;if (IsBalanceTree(_root->_left, left) && IsBalanceTree(_root->_right, right)){int diff = left - right;//定义一个数值等于左边减去右边if (diff < 1 && diff >= -1){Depth = 1 + (left>right ? left : right);return true;}}return false;}