在C++中,如果你需要快速获取矩阵中一个位置(如(i, j))四周的值,可以使用一个方向数组(或方向向量)来简化代码。方向数组定义了目标点的上下左右相对偏移量,配合循环可以避免重复代码。

以下是一个示例代码:

#include <iostream>
#include <vector>
using namespace std;

void getNeighbors(const vector<vector<int>>&amp; matrix, int x, int y) {
    // 定义方向数组:上、下、左、右
    int dx[4] = {-1, 1, 0, 0};
    int dy[4] = {0, 0, -1, 1};

    int rows = matrix.size();
    int cols = matrix[0].size();

    cout << &quot;Position (&quot; << x << &quot;, &quot; << y << &quot;) neighbors: &quot;;
    for (int k = 0; k < 4; ++k) {
        int nx = x + dx[k];
        int ny = y + dy[k];
        // 检查边界条件
        if (nx >= 0 &amp;&amp; nx < rows &amp;&amp; ny >= 0 &amp;&amp; ny < cols) {
            cout << matrix[nx][ny] << &quot; &quot;;
        }
    }
    cout << endl;
}

int main() {
    vector<vector<int>> matrix = {
        {1, 2, 3},
        {4, 5, 6},
        {7, 8, 9}
    };

    int x = 1, y = 1; // 中心位置
    getNeighbors(matrix, x, y);

    return 0;
}

运行结果
对于matrix矩阵:

1 2 3
4 5 6
7 8 9

如果查询(1, 1)位置的四周,输出结果为:

Position (1, 1) neighbors: 2 8 4 6

代码说明
方向数组:dx和dy表示相对偏移量,dx表示行的变化,dy表示列的变化。组合后可以生成上下左右的位置:上:(x - 1, y)下:(x + 1, y)左:(x, y - 1)右:(x, y + 1)
边界检查:通过nx >= 0 &amp;&amp; nx < rows确保行索引在合法范围内。通过ny >= 0 &amp;&amp; ny < cols确保列索引在合法范围内。
代码通用性:这种方法适用于任意大小的矩阵。如果需要扩展到八个方向(包括对角线),可以扩展方向数组为:
这种方式不仅代码简洁,而且易于扩展。

作者:徐子昂_2404020114_网络24_1
链接:https://www.nowcoder.com/discuss/694188703584968704?sourceSSR=users
来源:牛客网
全部评论

相关推荐

10-22 12:03
山东大学 Java
程序员小白条:26届一般都得有实习,项目可以随便写的,如果不是开源社区的项目,随便包装,技术栈也是一样,所以本质应该找学历厂,多投投央国企和银行,技术要求稍微低一点的,或者国企控股那种,纯互联网一般都得要干活
应届生简历当中,HR最关...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务