日志16

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

以下是一个示例代码：

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

void getNeighbors(const vector<vector<int>>& 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 << "Position (" << x << ", " << y << ") neighbors: ";
    for (int k = 0; k < 4; ++k) {
        int nx = x + dx[k];
        int ny = y + dy[k];
        // 检查边界条件
        if (nx >= 0 && nx < rows && ny >= 0 && ny < cols) {
            cout << matrix[nx][ny] << " ";
        }
    }
    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

代码说明

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

这种方式不仅代码简洁，而且易于扩展。