LeetCode in Kotlin

3248. Snake in Matrix

Easy

There is a snake in an n x n matrix grid and can move in four possible directions. Each cell in the grid is identified by the position: grid[i][j] = (i * n) + j.

The snake starts at cell 0 and follows a sequence of commands.

You are given an integer n representing the size of the grid and an array of strings commands where each command[i] is either "UP", "RIGHT", "DOWN", and "LEFT". It’s guaranteed that the snake will remain within the grid boundaries throughout its movement.

Return the position of the final cell where the snake ends up after executing commands.

Example 1:

Input: n = 2, commands = [“RIGHT”,”DOWN”]

Output: 3

Explanation:

image

Example 2:

Input: n = 3, commands = [“DOWN”,”RIGHT”,”UP”]

Output: 1

Explanation:

image

Constraints:

Solution

class Solution {
    fun finalPositionOfSnake(n: Int, commands: List<String>): Int {
        var x = 0
        var y = 0
        for (command in commands) {
            when (command) {
                "UP" -> if (x > 0) {
                    x--
                }
                "DOWN" -> if (x < n - 1) {
                    x++
                }
                "LEFT" -> if (y > 0) {
                    y--
                }
                "RIGHT" -> if (y < n - 1) {
                    y++
                }
            }
        }
        return (x * n) + y
    }
}