LeetCode in Kotlin

1970. Last Day Where You Can Still Cross

Hard

There is a 1-based binary matrix where 0 represents land and 1 represents water. You are given integers row and col representing the number of rows and columns in the matrix, respectively.

Initially on day 0, the entire matrix is land. However, each day a new cell becomes flooded with water. You are given a 1-based 2D array cells, where cells[i] = [ri, ci] represents that on the ith day, the cell on the rith row and cith column (1-based coordinates) will be covered with water (i.e., changed to 1).

You want to find the last day that it is possible to walk from the top to the bottom by only walking on land cells. You can start from any cell in the top row and end at any cell in the bottom row. You can only travel in the four cardinal directions (left, right, up, and down).

Return the last day where it is possible to walk from the top to the bottom by only walking on land cells.

Example 1:

Input: row = 2, col = 2, cells = [[1,1],[2,1],[1,2],[2,2]]

Output: 2

Explanation: The above image depicts how the matrix changes each day starting from day 0. The last day where it is possible to cross from top to bottom is on day 2.

Example 2:

Input: row = 2, col = 2, cells = [[1,1],[1,2],[2,1],[2,2]]

Output: 1

Explanation: The above image depicts how the matrix changes each day starting from day 0. The last day where it is possible to cross from top to bottom is on day 1.

Example 3:

Input: row = 3, col = 3, cells = [[1,2],[2,1],[3,3],[2,2],[1,1],[1,3],[2,3],[3,2],[3,1]]

Output: 3

Explanation: The above image depicts how the matrix changes each day starting from day 0. The last day where it is possible to cross from top to bottom is on day 3.

Constraints:

Solution

@Suppress("NAME_SHADOWING")
class Solution {
    fun latestDayToCross(row: Int, col: Int, cells: Array<IntArray>): Int {
        val ends = Array(row) { arrayOfNulls<Ends>(col) }
        for (i in cells.indices) {
            val r = cells[i][0] - 1
            val c = cells[i][1] - 1
            var curr: Ends? = null
            if (c > 0 && ends[r][c - 1] != null) {
                curr = calEnds(ends[r][c - 1], curr, c)
            }
            if (r > 0 && ends[r - 1][c] != null) {
                curr = calEnds(ends[r - 1][c], curr, c)
            }
            if (c < col - 1 && ends[r][c + 1] != null) {
                curr = calEnds(ends[r][c + 1], curr, c)
            }
            if (r < row - 1 && ends[r + 1][c] != null) {
                curr = calEnds(ends[r + 1][c], curr, c)
            }
            if (c > 0 && r > 0 && ends[r - 1][c - 1] != null) {
                curr = calEnds(ends[r - 1][c - 1], curr, c)
            }
            if (c > 0 && r < row - 1 && ends[r + 1][c - 1] != null) {
                curr = calEnds(ends[r + 1][c - 1], curr, c)
            }
            if (c < col - 1 && r > 0 && ends[r - 1][c + 1] != null) {
                curr = calEnds(ends[r - 1][c + 1], curr, c)
            }
            if (c < col - 1 && r < row - 1 && ends[r + 1][c + 1] != null) {
                curr = calEnds(ends[r + 1][c + 1], curr, c)
            }
            if (curr == null) {
                curr = Ends(i, c, c)
            }
            if (curr.l == 0 && curr.r == col - 1) {
                return i
            }
            ends[r][c] = curr
        }
        return cells.size
    }

    private fun calEnds(p: Ends?, curr: Ends?, c: Int): Ends? {
        var p = p
        var curr = curr
        while (p!!.parent != null) {
            p = p.parent
        }
        p.l = if (curr == null) Math.min(p.l, c) else Math.min(p.l, curr.l)
        p.r = if (curr == null) Math.max(p.r, c) else Math.max(p.r, curr.r)
        if (curr == null) {
            curr = p
        } else if (curr.i != p.i) {
            curr.parent = p
            curr = curr.parent
        }
        return curr
    }

    internal class Ends(var i: Int, var l: Int, var r: Int) {
        var parent: Ends? = null
    }
}