diff options
author | crupest <crupest@outlook.com> | 2021-09-26 12:50:45 +0800 |
---|---|---|
committer | crupest <crupest@outlook.com> | 2021-09-26 12:50:45 +0800 |
commit | cf2cea7e417a9d2bc946587d4a6ce48d6d4403d8 (patch) | |
tree | 6d08eea35a9f1cd9a66da48fbc2e5e88f742100e /leetcode/lccup/2021/1.cpp | |
parent | 450e391b6feb6e6dcefd07a3b85dfd23b69fa283 (diff) | |
download | solutions-cf2cea7e417a9d2bc946587d4a6ce48d6d4403d8.tar.gz solutions-cf2cea7e417a9d2bc946587d4a6ce48d6d4403d8.tar.bz2 solutions-cf2cea7e417a9d2bc946587d4a6ce48d6d4403d8.zip |
...
Diffstat (limited to 'leetcode/lccup/2021/1.cpp')
-rw-r--r-- | leetcode/lccup/2021/1.cpp | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/leetcode/lccup/2021/1.cpp b/leetcode/lccup/2021/1.cpp new file mode 100644 index 0000000..550da04 --- /dev/null +++ b/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 |