日期: 2024 年 11 月 21 日

目标

大小为 n x n 的矩阵 grid 中有一条蛇。蛇可以朝 四个可能的方向 移动。矩阵中的每个单元格都使用位置进行标识： grid[i][j] = (i * n) + j。

蛇从单元格 0 开始，并遵循一系列命令移动。

给你一个整数 n 表示 grid 的大小，另给你一个字符串数组 commands，其中包括 "UP"、"RIGHT"、"DOWN" 和 "LEFT"。题目测评数据保证蛇在整个移动过程中将始终位于 grid 边界内。

返回执行 commands 后蛇所停留的最终单元格的位置。

示例 1：

输入：n = 2, commands = ["RIGHT","DOWN"]
输出：3

示例 2：

输入：n = 3, commands = ["DOWN","RIGHT","UP"]
输出：1

说明：

• 2 <= n <= 10
• 1 <= commands.length <= 100
• commands 仅由 "UP"、"RIGHT"、"DOWN" 和 "LEFT" 组成。
• 生成的测评数据确保蛇不会移动到矩阵的边界外。

思路

有一个 n x n 矩阵 grid，初始时位置 (0, 0) 有条蛇，有一系列命令可以操作蛇移到，操作保证在矩阵内移动，问蛇最后停留的位置，格子的标识为 grid[i][j] = (i * n) + j。

直接模拟操作就可以了，将操作映射为行列的增减，直接计算位置即可。

最快的解法仅比较操作的第一个字符，并且上下移动直接减加 n，最后不用乘法计算。

代码

/**
 * @date 2024-11-21 0:39
 */
public class FinalPositionOfSnake3248 {

    /**
     * 最快题解
     */
    class Solution {
        public int finalPositionOfSnake(int n, List<String> commands) {
            int ans = 0;
            for (String c : commands) {
                if (c.charAt(0) == 'U') {
                    ans -= n;
                } else if (c.charAt(0) == 'D') {
                    ans += n;
                } else if (c.charAt(0) == 'L') {
                    --ans;
                } else {
                    ++ans;
                }
            }
            return ans;
        }
    }

    public static Map<String, int[]> map = new HashMap<>(4);

    static {
        map.put("UP", new int[]{-1, 0});
        map.put("RIGHT", new int[]{0, 1});
        map.put("DOWN", new int[]{1, 0});
        map.put("LEFT", new int[]{0, -1});
    }

    public int finalPositionOfSnake(int n, List<String> commands) {
        int[] move = new int[2];
        for (String command : commands) {
            move[0] += map.get(command)[0];
            move[1] += map.get(command)[1];
        }
        return move[0] * n + move[1];
    }

}

性能