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 | ca5a0f8dde5d8d9d7f877a790d05a270cc3224f9 (patch) | |
| tree | 975d29261c8e33bebc3dd57ecf947bc74764a710 /works | |
| parent | d4d511df8c7543839b991f3a741337e34a48fd5f (diff) | |
| download | crupest-ca5a0f8dde5d8d9d7f877a790d05a270cc3224f9.tar.gz crupest-ca5a0f8dde5d8d9d7f877a790d05a270cc3224f9.tar.bz2 crupest-ca5a0f8dde5d8d9d7f877a790d05a270cc3224f9.zip | |
import(solutions): Add problem 543.
Diffstat (limited to 'works')
| -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);
+ }
+};
|