Skip to content

Commit 452ccd2

Browse files
committed
Added test
1 parent 1b76ef8 commit 452ccd2

File tree

2 files changed

+10
-10
lines changed
  • src
    • main/kotlin/g3601_3700/s3608_minimum_time_for_k_connected_components
    • test/kotlin/g3601_3700/s3608_minimum_time_for_k_connected_components

2 files changed

+10
-10
lines changed

src/main/kotlin/g3601_3700/s3608_minimum_time_for_k_connected_components/Solution.kt

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -28,24 +28,16 @@ class Solution {
2828

2929
private fun countComponents(n: Int, edges: Array<IntArray>, t: Int): Int {
3030
val parent = IntArray(n)
31-
val size = IntArray(n)
3231
for (i in 0..<n) {
3332
parent[i] = i
34-
size[i] = 1
3533
}
3634
var comps = n
3735
for (e in edges) {
3836
if (e[2] > t) {
39-
var u = find(parent, e[0])
40-
var v = find(parent, e[1])
37+
val u = find(parent, e[0])
38+
val v = find(parent, e[1])
4139
if (u != v) {
42-
if (size[u] < size[v]) {
43-
val tmp = u
44-
u = v
45-
v = tmp
46-
}
4740
parent[v] = u
48-
size[u] += size[v]
4941
comps--
5042
}
5143
}

src/test/kotlin/g3601_3700/s3608_minimum_time_for_k_connected_components/SolutionTest.kt

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,4 +32,12 @@ internal class SolutionTest {
3232
equalTo<Int>(0),
3333
)
3434
}
35+
36+
@Test
37+
fun minTime4() {
38+
assertThat(
39+
Solution().minTime(3, arrayOf<IntArray>(intArrayOf(2, 1, 1469), intArrayOf(1, 0, 5701)), 2),
40+
equalTo(1469),
41+
)
42+
}
3543
}

0 commit comments

Comments
 (0)