2025-03-20：矩阵中的蛇。用go语言，给定一个大小为 n x n 的矩阵 grid，其中有一条蛇。蛇可以向四个方向移动：

福大大架构师每日一题

发布于 2025-03-20 18:50:01

2320

2025-03-20：矩阵中的蛇。用go语言，给定一个大小为 n x n 的矩阵 grid，其中有一条蛇。蛇可以向四个方向移动：上、右、下和左。矩阵的每个单元格由其位置标识，公式为 grid[i][j] = (i * n) + j。

蛇从单元格 0 开始，并根据给定的一系列命令进行移动。您将获得一个整数 n，表示矩阵的大小，以及一个字符串数组 commands，其中包含指令如 "UP"、"RIGHT"、"DOWN" 和 "LEFT"。题目保证在整个移动过程中蛇始终留在矩阵边界内。

请编写代码，返回执行完所有指令后，蛇最终停留在的单元格位置。

2 <= n <= 10。

1 <= commands.length <= 100。

commands 仅由 "UP"、"RIGHT"、"DOWN" 和 "LEFT" 组成。

生成的测评数据确保蛇不会移动到矩阵的边界外。

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

输出：3。解释：

题目来自leetcode3248。

大体步骤如下：

1.初始化一个变量 ans 为 0，用于表示蛇在矩阵中的位置。开始位置为单元格 0。

2.对于提供的每一个指令 c：

• 如果指令以 'U' 开头，表示向上移动，即蛇的位置减去当前行数 n。
• 如果指令以 'D' 开头，表示向下移动，即蛇的位置加上当前行数 n。
• 如果指令以 'L' 开头，表示向左移动，即蛇的位置减1。
• 如果指令以 'R' 开头，表示向右移动，即蛇的位置加1。

3.返回最终的位置 ans。

总体而言，对每个指令进行处理，最后返回蛇的最终停留位置。时间复杂度为 O(n)，其中 n 为指令的数量，额外空间复杂度为 O(1)，因为只使用了常数级的额外空间。

Go完整代码如下：

package main

import (
    "fmt"
)

func finalPositionOfSnake(n int, commands []string)int {
    ans := 0
    for _, c := range commands {
        switch c[0] {
        case'U':
            ans -= n
        case'D':
            ans += n
        case'L':
            ans--
        case'R':
            ans++
        }
    }
    return ans
}

func main() {
    n := 2
    commands := []string{"RIGHT", "DOWN"}
    result := finalPositionOfSnake(n, commands)
    fmt.Println(result)
}

Python完整代码如下：

# -*-coding:utf-8-*-

deffinal_position_of_snake(n, commands):
    ans = 0
    for c in commands:
        if c[0] == 'U':
            ans -= n
        elif c[0] == 'D':
            ans += n
        elif c[0] == 'L':
            ans -= 1
        elif c[0] == 'R':
            ans += 1
    return ans

if __name__ == "__main__":
    n = 2
    commands = ["RIGHT", "DOWN"]
    result = final_position_of_snake(n, commands)
    print(result)