diff options
| author | crupest <crupest@outlook.com> | 2021-02-23 17:36:58 +0800 |
|---|---|---|
| committer | crupest <crupest@outlook.com> | 2021-02-23 17:36:58 +0800 |
| commit | 7e3e89a3f0ecc49c33334f423dbde3a5fdc57abd (patch) | |
| tree | 93589027113c19623b74faf8d131ba0636a9bd12 | |
| parent | f444648a2fe6eb84a9aeb4f7c7c4e09020475bf2 (diff) | |
| download | crupest-7e3e89a3f0ecc49c33334f423dbde3a5fdc57abd.tar.gz crupest-7e3e89a3f0ecc49c33334f423dbde3a5fdc57abd.tar.bz2 crupest-7e3e89a3f0ecc49c33334f423dbde3a5fdc57abd.zip | |
import(solutions): Add problem 543.
| -rw-r--r-- | works/solutions/cpp/543.cpp | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/works/solutions/cpp/543.cpp b/works/solutions/cpp/543.cpp new file mode 100644 index 0000000..f782521 --- /dev/null +++ b/works/solutions/cpp/543.cpp @@ -0,0 +1,32 @@ +struct TreeNode {
+ int val;
+ TreeNode *left;
+ TreeNode *right;
+ TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
+};
+
+#include <algorithm>
+
+class Solution {
+public:
+ int diameterOfBinaryTree(TreeNode *root) {
+ int max = 0;
+ depth(root, max);
+ return max;
+ }
+
+ static int depth(TreeNode *root, int &max) {
+ if (root == nullptr)
+ return -1;
+
+ auto left = depth(root->left, max) + 1;
+ auto right = depth(root->right, max) + 1;
+
+ auto current_max = left + right;
+ if (current_max > max) {
+ max = current_max;
+ }
+
+ return std::max(left, right);
+ }
+};
|