Skip to content

Commit 7eb35eb

Browse files
authored
feat: add swift implementation to lcp problem: No.44 (#3784)
1 parent b049df8 commit 7eb35eb

File tree

2 files changed

+59
-0
lines changed

2 files changed

+59
-0
lines changed

lcp/LCP 44. 开幕式焰火/README.md

+32
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,38 @@ func dfs(root *TreeNode) {
159159
}
160160
```
161161

162+
#### Swift
163+
164+
```swift
165+
/* class TreeNode {
166+
* var val: Int
167+
* var left: TreeNode?
168+
* var right: TreeNode?
169+
* init(_ val: Int) {
170+
* self.val = val
171+
* self.left = nil
172+
* self.right = nil
173+
* }
174+
* }
175+
*/
176+
177+
class Solution {
178+
private var uniqueColors: Set<Int> = []
179+
180+
func numColor(_ root: TreeNode?) -> Int {
181+
dfs(root)
182+
return uniqueColors.count
183+
}
184+
185+
private func dfs(_ node: TreeNode?) {
186+
guard let node = node else { return }
187+
uniqueColors.insert(node.val)
188+
dfs(node.left)
189+
dfs(node.right)
190+
}
191+
}
192+
```
193+
162194
<!-- tabs:end -->
163195

164196
<!-- solution:end -->
+27
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
/* class TreeNode {
2+
* var val: Int
3+
* var left: TreeNode?
4+
* var right: TreeNode?
5+
* init(_ val: Int) {
6+
* self.val = val
7+
* self.left = nil
8+
* self.right = nil
9+
* }
10+
* }
11+
*/
12+
13+
class Solution {
14+
private var uniqueColors: Set<Int> = []
15+
16+
func numColor(_ root: TreeNode?) -> Int {
17+
dfs(root)
18+
return uniqueColors.count
19+
}
20+
21+
private func dfs(_ node: TreeNode?) {
22+
guard let node = node else { return }
23+
uniqueColors.insert(node.val)
24+
dfs(node.left)
25+
dfs(node.right)
26+
}
27+
}

0 commit comments

Comments
 (0)