题目：
给你一个 n x n 的二维数组 grid，它包含范围 [0, n2 - 1] 内的不重复元素。
实现 neighborSum 类：
neighborSum(int [][]grid)初始化对象。int adjacentSum(int value)返回在grid中与value相邻的元素之和，相邻指的是与value在上、左、右或下的元素。int diagonalSum(int value)返回在grid中与value对角线相邻的元素之和，对角线相邻指的是与value在左上、右上、左下或右下的元素。
示例 1：
输入：
[“neighborSum”, “adjacentSum”, “adjacentSum”, “diagonalSum”, “diagonalSum”]
[[[[0, 1, 2], [3, 4, 5], [6, 7, 8]]], [1], [4], [4], [8]]
输出： [null, 6, 16, 16, 4]
解释：
- 1 的相邻元素是 0、2 和 4。
- 4 的相邻元素是 1、3、5 和 7。
- 4 的对角线相邻元素是 0、2、6 和 8。
- 8 的对角线相邻元素是 4。
示例 2：
输入：
[“neighborSum”, “adjacentSum”, “diagonalSum”]
[[[[1, 2, 0, 3], [4, 7, 15, 6], [8, 9, 10, 11], [12, 13, 14, 5]]], [15], [9]]
输出： [null, 23, 45]
解释：
- 15 的相邻元素是 0、10、7 和 6。
- 9 的对角线相邻元素是 4、12、14 和 15。
提示：
3 <= n == grid.length == grid[0].length <= 100 <= grid[i][j] <= n2 - 1- 所有
grid[i][j]值均不重复。 adjacentSum和diagonalSum中的value均在范围[0, n2 - 1]内。- 最多会调用
adjacentSum和diagonalSum总共2 * n2次。
思路：
先用哈希表做索引，将每个 value 对应到 grid 的行列索引存下来，然后判断周围有没有数字即可
代码：
class NeighborSum {
public:
vector<vector<int>> grid;
int n;
unordered_map<int, pair<int, int>> hash;
NeighborSum(vector<vector<int>>& grid) {
this->grid = grid;
n = grid.size();
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
hash[grid[i][j]] = make_pair(i, j);
}
}
}
int adjacentSum(int value) {
// 上下左右
auto p = hash[value];
int row = p.first, col = p.second;
int ans = 0;
int up = 0, down = 0, left = 0, right = 0;
if (row - 1 >= 0) left = grid[row - 1][col];
if (row + 1 < n) right = grid[row + 1][col];
if (col - 1 >= 0) up = grid[row][col - 1];
if (col + 1 < n) down = grid[row][col + 1];
return left + right + up + down;
}
int diagonalSum(int value) {
// 斜上下左右
auto p = hash[value];
int row = p.first, col = p.second;
int ans = 0;
int up = 0, down = 0, left = 0, right = 0;
if (row - 1 >= 0 && col - 1 >= 0) left = grid[row - 1][col - 1];
if (row + 1 < n && col - 1 >= 0) right = grid[row + 1][col - 1];
if (col + 1 < n && row - 1 >= 0) up = grid[row - 1][col + 1];
if (col + 1 < n && row + 1 < n) down = grid[row + 1][col + 1];
return left + right + up + down;
}
};
/**
* Your NeighborSum object will be instantiated and called as such:
* NeighborSum* obj = new NeighborSum(grid);
* int param_1 = obj->adjacentSum(value);
* int param_2 = obj->diagonalSum(value);
*/