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