二叉树:二叉树的直径

2022-02-24 19:40:47 浏览数 (1)

二叉树直径的定义:二叉树中路径的最大长度 二叉树中路径的最大长度,可以理解所有节点的左右子树高度之和的最大值。假设二叉树有n个节点,编号为{a1,a2,…,an}, 其对应的左右子树的高度之和为H = {h1,h2,h3,…,hn}, 则该二叉树的直径为max(H)。

代码语言:javascript复制
struct node {
	int val;
	node *left, *right;
};

int ans  = 0;// 用于维护任意节点左右子树高度和的最大值
//dfs()函数的返回值是当前节点的高度

int dfs(node *root) {
	if (!root) return 0;
	int left = dfs(root->left);
	int right = dfs(root->right);
	ans = max(ans, left right);
	return max(right, left)   1;
}


int diameterOfBinaryTree(node *root) {
	dfs(root);
	return ans;
}

0 人点赞