From 185ef9fcb0e59f13e9ee0ccb261693cdaddebab0 Mon Sep 17 00:00:00 2001 From: crupest Date: Tue, 23 Feb 2021 21:07:19 +0800 Subject: import(solutions): Move leetcode solutions to subdir. --- works/solutions/leetcode/cpp/543.cpp | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 works/solutions/leetcode/cpp/543.cpp (limited to 'works/solutions/leetcode/cpp/543.cpp') diff --git a/works/solutions/leetcode/cpp/543.cpp b/works/solutions/leetcode/cpp/543.cpp new file mode 100644 index 0000000..f782521 --- /dev/null +++ b/works/solutions/leetcode/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 + +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