Leetcode 矩阵中的幻方

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

链接：https://leetcode-cn.com/problems/magic-squares-in-grid

 

 

思路：暴力法，注意三阶幻方的定义是1到9，且不重复，横竖斜之和相等。

 

bool isMagicSquares(int **g, int x, int y) {
    
    int i ,j;
    int s;
    char c[9] = {0};
    
    for(i = 0; i <= 2; i++) {
        for(j = 0; j <= 2; j++) {
            if(g[x+i][y+j] > 9 || g[x+i][y+j] < 1)
                return false;
            else
                c[g[x+i][y+j]-1] = 1;
       }
    }
    
    if(c[0] + c[1] + c[2] + c[3] + c[4] + c[5] + c[6] + c[7] + c[8] != 9)
        return false;
    
    s = g[x][y] + g[x][y+1] + g[x][y+2];
    
    if(s != g[x+1][y] + g[x+1][y+1] + g[x+1][y+2])
        return false;
    
    if(s != g[x+2][y] + g[x+2][y+1] + g[x+2][y+2])
        return false;
    
    if(s != g[x][y] + g[x+1][y] + g[x+2][y])
        return false;
    
    if(s != g[x][y+1] + g[x+1][y+1] + g[x+2][y+1])
        return false;
    
    if(s != g[x][y+2] + g[x+1][y+2] + g[x+2][y+2])
        return false;
    
    if(s != g[x][y] + g[x+1][y+1] + g[x+2][y+2])
        return false;
    
    if(s != g[x][y+2] + g[x+1][y+1] + g[x+2][y])
        return false;
    
    return true;
}

int numMagicSquaresInside(int** grid, int gridSize, int* gridColSize){
    
    int i, j;
    int ret = 0;

    
    for(i = 0; i <= gridSize -3; i++) 
        for(j = 0; j <= *gridColSize -3; j++) 
            if(isMagicSquares(grid,i,j)) 
                ret++;
            
    
    return ret;
}