二叉树直径的定义:二叉树中路径的最大长度 二叉树中路径的最大长度,可以理解所有节点的左右子树高度之和的最大值。假设二叉树有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;
}