题目(简单)
解题思路
单独考虑一个根节点的直径,是代表左边深度+右边深度,dfs求深度即可;
考虑孩子作为中间点可能产生更大的直径;
那一棵树的直径就是自己的直径、左孩直径、右孩直径中最大值;
套递归就行了
代码
class Solution {
public:
int dfs(TreeNode* root, int len) { //树的深度
if(root == nullptr) return len;
return max(dfs(root->left, len+1), dfs(root->right, len+1));
}
int diameterOfBinaryTree(TreeNode* root) {
if(root == nullptr) return 0;
int ans = dfs(root->left, 0) + dfs(root->right, 0);
return max(ans, max(diameterOfBinaryTree(root->left), diameterOfBinaryTree(root->right)));
}
};