题目描述:
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
题解:
class Solution
{
public:
int ans;
int diameterOfBinaryTree(TreeNode* root)
{
ans = 1;
dfs(root);
return ans-1; //深度等于节点数减一
}
int dfs(TreeNode* rt)
{
if(root == NULL)
return 0;
int l = dfs(rt->left);
int r = dfs(rt->right);
ans = max(ans,l+r+1);
return max(l,r)+1;
}
};
解题思路:
根据题意,二叉树的半径是任意两个点之间的最大长度,那么利用后序遍历,对于最长路径的更新就是一个节点的左右节点加本身的长度。
而对于一个节点,它的返回值,为其与左节点或者右节点 +1 的最大值。