From 1f9626144c4856d0a7d93b125175879d5a6685d4 Mon Sep 17 00:00:00 2001 From: crupest Date: Sun, 26 Sep 2021 12:50:45 +0800 Subject: import(solutions): ... --- works/solutions/leetcode/lccup/2021/1.cpp | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 works/solutions/leetcode/lccup/2021/1.cpp (limited to 'works/solutions/leetcode/lccup/2021/1.cpp') 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 -- cgit v1.2.3