数据结构 - 求二叉树中结点的最大距离（C++）

最新推荐文章于 2022-10-29 12:19:15 发布

weixin_33704591

最新推荐文章于 2022-10-29 12:19:15 发布

阅读量203

点赞数

文章标签： c/c++ 数据结构与算法

本文详细介绍了B树节点长度计算的实现方法，包括模板类的使用、递归查找子树最大长度以及如何更新根节点的最大长度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

// ------BTreeMaxNodeLength.cpp------

#include <iostream>

using namespace std;

template <class T>
struct BTNode
{
	// 左孩子
	BTNode<T> *lChild;
	// 右孩子
	BTNode<T> *rChild;
	// 该结点的值
	T data;
	// 左子树最长距离
	int leftSubTreeMaxLength;
	// 右子树最长距离
	int rightSubTreeMaxLength;
};

template <class T>
class BinaryTree
{
public:
	void GetMaxNodeLength(BTNode<T> * root, int *maxNodeLength)
	{
		// 遍历到叶子结点，返回
		if (root == NULL)
		{
			return;
		}

		// 假设左子树为空，那么该结点左子树最长距离为0
		if (root->lChild == NULL)
		{
			root->leftSubTreeMaxLength = 0;
		}

		// 假设右子树为空，那么该结点右子树最长距离为0
		if (root->rChild == NULL)
		{
			root->rightSubTreeMaxLength = 0;
		}

		// 假设左子树不为空，递归查找左子树最长距离
		if (root->lChild != NULL)
		{
			GetMaxNodeLength(root->lChild, maxNodeLength);
		}

		// 假设右子树不为空，递归查找右子树最长距离
		if (root->rChild != NULL)
		{
			GetMaxNodeLength(root->rChild, maxNodeLength);
		}

		// 计算左子树中距离根结点的最长距离
		if (root->lChild != NULL)
		{
			if (root->lChild->leftSubTreeMaxLength > root->lChild->rightSubTreeMaxLength)
			{
				root->leftSubTreeMaxLength = root->lChild->leftSubTreeMaxLength + 1;
			}
			else
			{
				root->leftSubTreeMaxLength = root->lChild->rightSubTreeMaxLength + 1;
			}
		}

		// 计算右子树中距离根结点的最长距离
		if (root->rChild != NULL)
		{
			if (root->rChild->leftSubTreeMaxLength > root->rChild->rightSubTreeMaxLength)
			{
				root->rightSubTreeMaxLength = root->rChild->leftSubTreeMaxLength + 1;
			}
			else
			{
				root->rightSubTreeMaxLength = root->rChild->rightSubTreeMaxLength + 1;
			}
		}

		// 更新最长距离
		if (root->leftSubTreeMaxLength + root->rightSubTreeMaxLength > *maxNodeLength)
		{
			*maxNodeLength = root->leftSubTreeMaxLength + root->rightSubTreeMaxLength;
		}
	}
};