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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/magic-squares-in-grid
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

这道题就是一个体力活,把每一行中横竖斜都拿出来判断一下是否符合要求即可。

若是符合要求,其中心必定是5.

 

    private int[][] grid;
    private int[] arr = new int[10];

    public int numMagicSquaresInside(int[][] grid) {
        this.grid = grid;
        int a = grid.length - 1;
        int b = grid[0].length - 1;
        int count = 0;


        for (int i = 1; i < a; i++) {
            for (int j = 1; j < b; j++) {
                boolean b1 = find(i, j);
                if (b1) {
                    count++;
                    j++;
                }
            }
        }
        return count;
    }

    private boolean find(int m, int n) {
        if (grid[m][n] != 5) {
            return false;
        }
        Arrays.fill(arr, 0);
        int a = grid[m][n - 1];
        if (a > 9 || a < 0) {
            return false;
        }
        arr[a] = 1;
        int b = grid[m][n + 1];
        if (b > 9 || b < 0) {
            return false;
        }
        if (arr[b] == 1) {
            return false;
        }
        if (a + b != 10) {
            return false;
        }
        int c = grid[m - 1][n - 1];
        if (c > 9 || c < 0) {
            return false;
        }
        if (arr[c] == 1) {
            return false;
        }
        int d = grid[m - 1][n];
        if (d > 9 || d < 0) {
            return false;
        }
        if (arr[d] == 1) {
            return false;
        }
        int e = grid[m - 1][n + 1];
        if (e > 9 || e < 0) {
            return false;
        }
        if (arr[e] == 1) {
            return false;
        }
        if (c + d + e != 15) {
            return false;
        }
        int f = grid[m + 1][n - 1];
        if (f > 9 || f < 0) {
            return false;
        }
        if (arr[f] == 1) {
            return false;
        }
        int g = grid[m + 1][n];
        if (g > 9 || g < 0) {
            return false;
        }
        if (arr[g] == 1) {
            return false;
        }
        int h = grid[m + 1][n + 1];
        if (h > 9 || h < 0) {
            return false;
        }
        if (arr[h] == 1) {
            return false;
        }
        if (f + g + h != 15) {
            return false;
        }
        if (c + a + f != 15) {
            return false;
        }
        if (e + b + h != 15) {
            return false;
        }
        if (d + g != 10) {
            return false;
        }
        if (c + h != 10) {
            return false;
        }
        return e + f == 10;
    }

posted @ 2021-08-15 18:00  旺仔古李  阅读(38)  评论(0编辑  收藏  举报