Update diameter-of-binary-tree.cpp

kamyu104 · web-flow · commit 7d07b50bb4d8 · 2018-08-05T14:22:38.000+08:00
diff --git a/C++/diameter-of-binary-tree.cpp b/C++/diameter-of-binary-tree.cpp
@@ -15,7 +15,7 @@ class Solution {
     int diameterOfBinaryTree(TreeNode* root) {
         int diameter = 1;
         depth(root, &diameter);
-        return diameter - 1;
+        return diameter;
     }
 
 private:
@@ -25,7 +25,7 @@ class Solution {
         }
         auto left = depth(root->left, diameter);
         auto right = depth(root->right, diameter);
-        *diameter = max(*diameter, 1 + left + right);
-        return 1 + max(left, right);
+        *diameter = max(*diameter, left + right);
+        return max(left, right);
     }
 };

Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,7 @@ class Solution {`
`15`	`15`	`int diameterOfBinaryTree(TreeNode* root) {`
`16`	`16`	`int diameter = 1;`
`17`	`17`	`depth(root, &diameter);`
`18`		`- return diameter - 1;`
	`18`	`+ return diameter;`
`19`	`19`	`}`
`20`	`20`
`21`	`21`	`private:`
`@@ -25,7 +25,7 @@ class Solution {`
`25`	`25`	`}`
`26`	`26`	`auto left = depth(root->left, diameter);`
`27`	`27`	`auto right = depth(root->right, diameter);`
`28`		`- diameter = max(diameter, 1 + left + right);`
`29`		`- return 1 + max(left, right);`
	`28`	`+ diameter = max(diameter, left + right);`
	`29`	`+ return max(left, right);`
`30`	`30`	`}`
`31`	`31`	`};`