Leetcode 840. 矩阵中的幻方 C++

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. 1 <= grid.length <= 10
2. 1 <= grid[0].length <= 10
3. 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。