forked from igorwojda/kotlin-coding-challenges
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathsolution.kt
34 lines (31 loc) · 920 Bytes
/
solution.kt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
package com.igorwojda.binarytree.insert
private object Solution1 {
private data class Node<E : Comparable<E>>(
var data: E,
var left: Node<E>? = null,
var right: Node<E>? = null
) {
fun insert(e: E) {
if (e < data) { // left node
if (left == null) {
left = Node(e)
} else {
left?.insert(e)
}
} else if (e > data) { // right node
if (right == null) {
right = Node(e)
} else {
right?.insert(e)
}
}
}
fun contains(e: E): Boolean = when {
e == data -> true
e < data -> left?.contains(e) ?: false
e > data -> right?.contains(e) ?: false
else -> false
}
}
}
private object KtLintWillNotComplain