求二叉树的高度

伊H2024-08-26 8:34

思路

利用二叉树的结构特点和递归思路，先求出左树高度和右树高度，较高的子树 + 1（根节点也占了一个高度单位）就是本树的高度。

代码

代码1，严重的计算重复问题

cpp 复制代码

int TreeHeight(BTNode* root)
{
	if (root == NULL)
		return 0;

	return TreeHeight(root->left) > TreeHeight(root->right)
		? TreeHeight(root->left) + 1 : TreeHeight(root->right) + 1;
}

比较时计算了一遍，返回时又计算了一遍，更可怕的是每颗子树都有这样的问题，虽然代码简单，但重复问题非常严重。

代码2，记录结果，避免重复计算

cpp 复制代码

int TreeHeight(BTNode* root)
{
	if (root == NULL)
		return 0;

	int leftHeight = TreeHeight(root->left);
	int rightHeight = TreeHeight(root->right);

	return leftHeight > rightHeight ? leftHeight + 1 : rightHeight + 1;
}

比较时计算了一遍，这时把结果记录到变量里，返回时直接使用变量返回，没有重复计算。

代码3，fmax函数

cpp 复制代码

int TreeHeight(BTNode* root)
{
	if (root == NULL)
		return 0;

	return fmax(TreeHeight(root->left), TreeHeight(root->right)) + 1;
}

实参完成计算过程，传递给形参，形参直接用作返回，计算过程不重复。