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