diff options
Diffstat (limited to 'works/solutions/cpp')
| -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);
 +  }
 +};
  | 
