File tree Expand file tree Collapse file tree 2 files changed +8
-8
lines changed
src/main/kotlin/g3501_3600
s3558_number_of_ways_to_assign_edge_weights_i
s3559_number_of_ways_to_assign_edge_weights_ii Expand file tree Collapse file tree 2 files changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -22,7 +22,7 @@ class Solution {
22
22
val q: Queue <IntArray > = LinkedList <IntArray >()
23
23
q.offer(intArrayOf(1 , 0 ))
24
24
l[1 ] = 0
25
- while (! q.isEmpty ()) {
25
+ while (q.isNotEmpty ()) {
26
26
val curr = q.peek()!! [0 ]
27
27
val level = q.peek()!! [1 ]
28
28
if (l[max] < l[curr]) {
Original file line number Diff line number Diff line change @@ -6,16 +6,16 @@ import kotlin.math.ceil
6
6
import kotlin.math.ln
7
7
8
8
class Solution {
9
- private var adj: MutableList <MutableList <Int ?> ? > ? = null
9
+ private lateinit var adj: MutableList <MutableList <Int >>
10
10
private lateinit var level: IntArray
11
11
private lateinit var jumps: Array <IntArray ?>
12
12
13
13
private fun mark (node : Int , par : Int ) {
14
- for (neigh in adj!! .get( node) !! ) {
14
+ for (neigh in adj[ node] ) {
15
15
if (neigh == par) {
16
16
continue
17
17
}
18
- level[neigh!! ] = level[node] + 1
18
+ level[neigh] = level[node] + 1
19
19
jumps[neigh]!! [0 ] = node
20
20
mark(neigh, node)
21
21
}
@@ -60,14 +60,14 @@ class Solution {
60
60
61
61
fun assignEdgeWeights (edges : Array <IntArray >, queries : Array <IntArray >): IntArray {
62
62
val n = edges.size + 1
63
- adj = ArrayList <MutableList <Int ?> ? > ()
63
+ adj = ArrayList <MutableList <Int > >()
64
64
level = IntArray (n)
65
65
for (i in 0 .. < n) {
66
- adj!! .add(ArrayList <Int ? >())
66
+ adj.add(ArrayList <Int >())
67
67
}
68
68
for (i in edges) {
69
- adj!! .get( i[0 ] - 1 ) !! .add(i[1 ] - 1 )
70
- adj!! .get( i[1 ] - 1 ) !! .add(i[0 ] - 1 )
69
+ adj[ i[0 ] - 1 ] .add(i[1 ] - 1 )
70
+ adj[ i[1 ] - 1 ] .add(i[0 ] - 1 )
71
71
}
72
72
val m = (ceil(ln(n - 1.0 ) / ln(2.0 ))).toInt() + 1
73
73
jumps = Array <IntArray ?>(n) { IntArray (m) }
You can’t perform that action at this time.
0 commit comments