【Python】【难度：简单】Leetcode 840. 矩阵中的幻方【完】

最新推荐文章于 2024-11-07 23:04:17 发布

原创

最新推荐文章于 2024-11-07 23:04:17 发布 · 432 阅读

0 ·

CC 4.0 BY-SA版权

给定一个3x3的整数矩阵grid，求解其中3x3的子矩阵中幻方的数量。幻方是行、列和对角线上的数字和都相等的矩阵。例如，输入矩阵grid有一个幻方子矩阵。题目要求输出grid中3x3的幻方子矩阵总数。

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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution(object):
    def numMagicSquaresInside(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        if len(grid) < 3 or len(grid[0]) < 3:
            return 0


        cnt = 0
        for i in range(len(grid) - 2):