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 {
2222 val q: Queue <IntArray > = LinkedList <IntArray >()
2323 q.offer(intArrayOf(1 , 0 ))
2424 l[1 ] = 0
25- while (! q.isEmpty ()) {
25+ while (q.isNotEmpty ()) {
2626 val curr = q.peek()!! [0 ]
2727 val level = q.peek()!! [1 ]
2828 if (l[max] < l[curr]) {
Original file line number Diff line number Diff line change @@ -6,16 +6,16 @@ import kotlin.math.ceil
66import kotlin.math.ln
77
88class Solution {
9- private var adj: MutableList <MutableList <Int ?> ? > ? = null
9+ private lateinit var adj: MutableList <MutableList <Int >>
1010 private lateinit var level: IntArray
1111 private lateinit var jumps: Array <IntArray ?>
1212
1313 private fun mark (node : Int , par : Int ) {
14- for (neigh in adj!! .get( node) !! ) {
14+ for (neigh in adj[ node] ) {
1515 if (neigh == par) {
1616 continue
1717 }
18- level[neigh!! ] = level[node] + 1
18+ level[neigh] = level[node] + 1
1919 jumps[neigh]!! [0 ] = node
2020 mark(neigh, node)
2121 }
@@ -60,14 +60,14 @@ class Solution {
6060
6161 fun assignEdgeWeights (edges : Array <IntArray >, queries : Array <IntArray >): IntArray {
6262 val n = edges.size + 1
63- adj = ArrayList <MutableList <Int ?> ? > ()
63+ adj = ArrayList <MutableList <Int > >()
6464 level = IntArray (n)
6565 for (i in 0 .. < n) {
66- adj!! .add(ArrayList <Int ? >())
66+ adj.add(ArrayList <Int >())
6767 }
6868 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 )
7171 }
7272 val m = (ceil(ln(n - 1.0 ) / ln(2.0 ))).toInt() + 1
7373 jumps = Array <IntArray ?>(n) { IntArray (m) }
You can’t perform that action at this time.
0 commit comments