思路:
多想想递归的本质。。多做做递归的题。还是做的太少了。
这题其实就是要想到一个转化的问题:题目要求的直径其实就是每个结点的左子树深度+右子树深度+1里的最大值。
可以回去看看递归求深度的dfs函数怎么写的,根据这个只需要加一个全局变量维护一下每个结点的左子树深度+右子树深度+1就可以了。
代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
int ans=0;
int dfs(TreeNode* root)
{
if(!root) return 0;
int ld=dfs(root->left);//左子树深度
int rd=dfs(root->right);//右子树深度
ans=max(ans,ld+rd+1);
return max(ld,rd)+1;
}
public:
int diameterOfBinaryTree(TreeNode* root) {
dfs(root);
return ans-1;
}
};