LeetCode 840. 矩阵中的幻方

目录结构

1.题目

2.题解


1.题目

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

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

2.题解

public class Solution840 {

    @Test
    public void test840() {
        int[][] grid = {
                {4, 3, 8, 4},
                {9, 5, 1, 9},
                {2, 7, 6, 2}};
        System.out.println(numMagicSquaresInside(grid));
    }

    public int numMagicSquaresInside(int[][] grid) {
        int result = 0;
        for (int i = 0; i <= grid.length - 3; i++) {
            for (int j = 0; j <= grid[0].length - 3; j++) {
                if (judge(grid, i, j)) {
                    result++;
                }
            }
        }
        return result;
    }

    public boolean judge(int[][] grid, int sR, int sC) {
        if (grid[sR + 1][sC + 1] != 5) {
            return false;
        }
        if (grid[sR][sC] + grid[sR + 1][sC + 1] + grid[sR + 2][sC + 2] != 15) {
            return false;
        }
        if (grid[sR][sC + 2] + grid[sR + 1][sC + 1] + grid[sR + 2][sC] != 15) {
            return false;
        }
        for (int j = sC; j < sC + 3; j++) {
            int count = 0;
            for (int i = sR; i < sR + 3; i++) {
                count += grid[i][j];
            }
            if (count != 15) {
                return false;
            }
        }
        Set<Integer> set = new HashSet<>();
        for (int i = sR; i < sR + 3; i++) {
            int count = 0;
            for (int j = sC; j < sC + 3; j++) {
                count += grid[i][j];
                if (1 <= grid[i][j] && grid[i][j] <= 9) {
                    set.add(grid[i][j]);
                }
            }
            if (count != 15) {
                return false;
            }
        }
        if (set.size() != 9) {
            return false;
        }
        return true;
    }
}
  • 时间复杂度:O(mn)
  • 空间复杂度:O(1)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值