LeetCode in Kotlin

529. Minesweeper

Medium

Let’s play the minesweeper game (Wikipedia, online game)!

You are given an m x n char matrix board representing the game board where:

You are also given an integer array click where click = [clickr, clickc] represents the next click position among all the unrevealed squares ('M' or 'E').

Return the board after revealing this position according to the following rules:

  1. If a mine 'M' is revealed, then the game is over. You should change it to 'X'.
  2. If an empty square 'E' with no adjacent mines is revealed, then change it to a revealed blank 'B' and all of its adjacent unrevealed squares should be revealed recursively.
  3. If an empty square 'E' with at least one adjacent mine is revealed, then change it to a digit ('1' to '8') representing the number of adjacent mines.
  4. Return the board when no more squares will be revealed.

Example 1:

Input: board = [[“E”,”E”,”E”,”E”,”E”],[“E”,”E”,”M”,”E”,”E”],[“E”,”E”,”E”,”E”,”E”],[“E”,”E”,”E”,”E”,”E”]], click = [3,0]

Output: [[“B”,”1”,”E”,”1”,”B”],[“B”,”1”,”M”,”1”,”B”],[“B”,”1”,”1”,”1”,”B”],[“B”,”B”,”B”,”B”,”B”]]

Example 2:

Input: board = [[“B”,”1”,”E”,”1”,”B”],[“B”,”1”,”M”,”1”,”B”],[“B”,”1”,”1”,”1”,”B”],[“B”,”B”,”B”,”B”,”B”]], click = [1,2]

Output: [[“B”,”1”,”E”,”1”,”B”],[“B”,”1”,”X”,”1”,”B”],[“B”,”1”,”1”,”1”,”B”],[“B”,”B”,”B”,”B”,”B”]]

Constraints:

Solution

class Solution {
    private var row = 0
    private var col = 0
    private fun dfs(board: Array<CharArray>, row: Int, col: Int) {
        if (row < 0 || row >= this.row || col < 0 || col >= this.col) {
            return
        }
        if (board[row][col] == 'E') {
            val numOfMine = bfs(board, row, col)
            if (numOfMine != 0) {
                board[row][col] = (numOfMine + '0'.code).toChar()
                return
            } else {
                board[row][col] = 'B'
            }
            for (i in DIRECTION) {
                dfs(board, row + i[0], col + i[1])
            }
        }
    }

    private fun bfs(board: Array<CharArray>, row: Int, col: Int): Int {
        var numOfMine = 0
        for (i in DIRECTION) {
            val newRow = row + i[0]
            val newCol = col + i[1]
            if (newRow >= 0 && newRow < this.row && newCol >= 0 && newCol < this.col && board[newRow][newCol] == 'M') {
                numOfMine++
            }
        }
        return numOfMine
    }

    fun updateBoard(board: Array<CharArray>, c: IntArray): Array<CharArray> {
        if (board[c[0]][c[1]] == 'M') {
            board[c[0]][c[1]] = 'X'
            return board
        } else {
            row = board.size
            col = board[0].size
            dfs(board, c[0], c[1])
        }
        return board
    }

    companion object {
        private val DIRECTION = arrayOf(
            intArrayOf(1, 0),
            intArrayOf(-1, 0),
            intArrayOf(0, 1),
            intArrayOf(0, -1),
            intArrayOf(-1, -1),
            intArrayOf(-1, 1),
            intArrayOf(1, -1),
            intArrayOf(1, 1)
        )
    }
}