这题在简单题中有点难度,主要是不要把边数和深度搞混(我就这样)。
我想了很久,发现如果当前节点没有右节点,就将它的右长度设为0,左节点同理,并且在递归是不会加一,而是将加一的操作放在有左右节点、从左右节点递归回来后。(而计算深度只需判断当前节点是否为空)
/*** 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 {
public:int result=0;int dg(TreeNode* root){int l=0;int r=0;if(root->left) l=dg(root->left)+1;if(root->right) r=dg(root->right)+1;result=max(result,l+r);return max(l,r);}int diameterOfBinaryTree(TreeNode* root){return max(result,dg(root));}
};
横向对比一下计算深度:
/*** 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 {
public:int result=0;int dg(TreeNode* root){if(root==nullptr) return 0;int r=dg(root->right);int l=dg(root->left);result=max(result,l+r);return max(l,r)+1;}int diameterOfBinaryTree(TreeNode* root){return max(result,dg(root));}
};