Leetcode 840. 矩阵中的幻方
题目
3 x 3 的幻方是一个填充有从 1 到 9 的不同数字的 3 x 3 矩阵,其中每行,每列以及两条对角线上的各数之和都相等。
给定一个由整数组成的 grid,其中有多少个 3 × 3 的 “幻方” 子矩阵?(每个子矩阵都是连续的)。
示例:
输入: [[4,3,8,4],
[9,5,1,9],
[2,7,6,2]]
输出: 1
解释:
下面的子矩阵是一个 3 x 3 的幻方:
438
951
276
而这一个不是:
384
519
762
总的来说,在本示例所给定的矩阵中只有一个 3 x 3 的幻方子矩阵。
提示:
- 1 <= grid.length <= 10
- 1 <= grid[0].length <= 10
- 0 <= grid[i][j] <=15
题解
枚举所有幻方可以知道,将其从上往下、从左往右构成共有8种可能,为81637492,83419672, 61873294, 67219834,49237816, 43891276, 29473618, 27691438。可见幻方的中心坐标一定为5,我们判断是否为这8种情况的一种,是则+1。详细过程见代码
代码
int numMagicSquaresInside(vector<vector<int>>& grid) {
int cnt=0,m = grid.size(),n = grid[0].size();
unordered_set<int> magicSquares{81637492, 83419672, 61873294, 67219834,
49237816, 43891276, 29473618, 27691438};
int dir[8][2] = {
{-1,-1},{-1,0},{-1,1},{0,-1},{0,1},{1,-1},{1,0},{1,1}
};
for(int i=1; i<m-1; i++){
for(int j=1; j<n-1; j++)
if(grid[i][j] == 5){
int sum = 0;
for(int k=0; k<8; k++)
sum = sum*10 + grid[i+dir[k][0]][j+dir[k][1]];
if(magicSquares.find(sum) != magicSquares.end()) cnt++;
}
}
return cnt;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/magic-squares-in-grid
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。