diff options
Diffstat (limited to 'works/solutions/leetcode/lccup')
| -rw-r--r-- | works/solutions/leetcode/lccup/2021/1.cpp | 27 | 
1 files changed, 27 insertions, 0 deletions
| diff --git a/works/solutions/leetcode/lccup/2021/1.cpp b/works/solutions/leetcode/lccup/2021/1.cpp new file mode 100644 index 0000000..550da04 --- /dev/null +++ b/works/solutions/leetcode/lccup/2021/1.cpp @@ -0,0 +1,27 @@ +struct TreeNode { +    int val; +    TreeNode *left; +    TreeNode *right; +    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} +}; + +class Solution { +public: +    int visited[1001] {0}; +    int result = 0; + +    void DFS(TreeNode* r) { +      if (!visited[r->val]) { +        result += 1; +        visited[r->val] = 1; +      } + +      if (r->left) DFS(r->left); +      if (r->right) DFS(r->right); +    } + +    int numColor(TreeNode* root) { +      DFS(root); +      return result; +    } +};
\ No newline at end of file | 
