您的位置:首页 > 理论基础 > 数据结构算法

(015)实现一个函数检查一棵树是否平衡(keep it up)

2014-08-29 02:41 393 查看
实现一个函数检查一棵树是否平衡。对于这个问题而言, 平衡指的是这棵树任意两个叶子结点到根结点的距离之差不大于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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  C++ 算法 数据结构
相关文章推荐