首页 > 代码库 > (015)实现一个函数检查一棵树是否平衡(keep it up)

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

实现一个函数检查一棵树是否平衡。对于这个问题而言, 平衡指的是这棵树任意两个叶子结点到根结点的距离之差不大于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;
}


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