实现一个函数检查一棵树是否平衡。对于这个问题而言, 平衡指的是这棵树任意两个叶子结点到根结点的距离之差不大于1。
这个题我们可以采用暴力搜索,找到叶子节点到根节点的最小值和最大值,然后他们的差如果大于1就不是平衡树,反之
则是平衡树。
int MinDepth = std::numeric_limits<int>::max();
int MaxDepth = std::numeric_limits<int>::min();
struct TreeNode
{
int data;
TreeNode* child;
TreeNode* brother;
};
bool isBlanceTree(const TreeNode* vNode, int vDepth=0)
{
if (vNode == NULL)
{
if (vDepth == 0) return true;
if (vDepth < MinDepth)
{
MinDepth = vDepth;
}
if (vDepth > MaxDepth)
{
MaxDepth = vDepth;
}
if (MaxDepth - MinDepth > 1) return false;
return true;
}
bool IsBlance = true;
if (vNode->child != NULL)
{
IsBlance = isBlanceTree(vNode->child, vDepth+1);
}
if (IsBlance && vNode->brother != NULL)
{
IsBlance = isBlanceTree(vNode->brother, vDepth);
}
return IsBlance;
}