LeetCode in Kotlin

2045. Second Minimum Time to Reach Destination

Hard

A city is represented as a bi-directional connected graph with n vertices where each vertex is labeled from 1 to n (inclusive). The edges in the graph are represented as a 2D integer array edges, where each edges[i] = [ui, vi] denotes a bi-directional edge between vertex ui and vertex vi. Every vertex pair is connected by at most one edge, and no vertex has an edge to itself. The time taken to traverse any edge is time minutes.

Each vertex has a traffic signal which changes its color from green to red and vice versa every change minutes. All signals change at the same time. You can enter a vertex at any time, but can leave a vertex only when the signal is green. You cannot wait at a vertex if the signal is green.

The second minimum value is defined as the smallest value strictly larger than the minimum value.

Given n, edges, time, and change, return the second minimum time it will take to go from vertex 1 to vertex n.

Notes:

Example 1:

       

Input: n = 5, edges = [[1,2],[1,3],[1,4],[3,4],[4,5]], time = 3, change = 5

Output: 13

Explanation:

The figure on the left shows the given graph.

The blue path in the figure on the right is the minimum time path.

The time taken is:

Hence the minimum time needed is 6 minutes.

The red path shows the path to get the second minimum time.

Hence the second minimum time is 13 minutes.

Example 2:

Input: n = 2, edges = [[1,2]], time = 3, change = 2

Output: 11

Explanation:

The minimum time path is 1 -> 2 with time = 3 minutes.

The second minimum time path is 1 -> 2 -> 1 -> 2 with time = 11 minutes.

Constraints:

Solution

import java.util.LinkedList
import java.util.Queue

class Solution {
    fun secondMinimum(n: Int, edges: Array<IntArray>, time: Int, change: Int): Int {
        val adj: Array<MutableList<Int>?> = arrayOfNulls(n)
        for (i in adj.indices) {
            adj[i] = ArrayList()
        }
        for (edge in edges) {
            val p = edge[0] - 1
            val q = edge[1] - 1
            adj[p]?.add(q)
            adj[q]?.add(p)
        }
        val dis1 = IntArray(n)
        val dis2 = IntArray(n)
        dis1.fill(Int.MAX_VALUE)
        dis2.fill(Int.MAX_VALUE)
        dis1[0] = 0
        val queue: Queue<IntArray> = LinkedList()
        queue.offer(intArrayOf(0, 0))
        while (queue.isNotEmpty()) {
            val temp = queue.poll()
            val cur = temp[0]
            val path = temp[1]
            for (node in adj[cur]!!) {
                val newPath = path + 1
                if (newPath < dis1[node]) {
                    dis2[node] = dis1[node]
                    dis1[node] = newPath
                    queue.offer(intArrayOf(node, newPath))
                } else if (newPath > dis1[node] && newPath < dis2[node]) {
                    dis2[node] = newPath
                    queue.offer(intArrayOf(node, newPath))
                }
            }
        }
        return helper(dis2[n - 1], time, change)
    }

    private fun helper(pathValue: Int, time: Int, change: Int): Int {
        var sum = 0
        for (i in 0 until pathValue) {
            sum += time
            if (i == pathValue - 1) {
                break
            }
            if (sum / change % 2 != 0) {
                sum = (sum / change + 1) * change
            }
        }
        return sum
    }
}