aboutsummaryrefslogtreecommitdiff
path: root/store/works/solutions/leetcode/lccup/2021/1.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'store/works/solutions/leetcode/lccup/2021/1.cpp')
-rw-r--r--store/works/solutions/leetcode/lccup/2021/1.cpp27
1 files changed, 27 insertions, 0 deletions
diff --git a/store/works/solutions/leetcode/lccup/2021/1.cpp b/store/works/solutions/leetcode/lccup/2021/1.cpp
new file mode 100644
index 0000000..550da04
--- /dev/null
+++ b/store/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