C++——二叉树的直径leetcode543

题目描述

给你一棵二叉树的根节点，返回该树的 直径 。
二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root 。
两节点之间路径的 长度 由它们之间边数表示。

思路

根据官方解答而来：
迭代，考虑写一个函数depth，输入二叉树中某个位置节点，返回以该节点为开始的最大长度，即向左或向右的最大深度，包括该节点。同时全局维护一个变量ans代表当前子树下的二叉树最大直径，则穿过该节点的二叉树最大直径为L+R，其中L为该节点左边不包括该节点的节点数，R为该节点右边不包括该节点的节点数。而ans=max(ans, L+R)

int ans=0;
int diameterOfBinaryTree(TreeNode* root)
{
	if(!root) return 0;
	depth(root);
	return ans;
}
int depth(TreeNode* root)
{
	if(!root) return 0;
	int L = depth(root->left);
	int R = depth(root->right);
	ans = max(ans, L+R);
	return max(L, R)+1;
}