From ca5a0f8dde5d8d9d7f877a790d05a270cc3224f9 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Tue, 23 Feb 2021 17:36:58 +0800
Subject: import(solutions): Add problem 543.

---
 works/solutions/cpp/543.cpp | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 works/solutions/cpp/543.cpp

(limited to 'works')

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);
+  }
+};
-- 
cgit v1.2.3