leetcode 840 .矩阵中的幻方

近期一直在用python刷leetcode的题，代码都提交到了github上了：https://github.com/Teingi/test

3 x 3 的幻方是一个填充有从 1 到 9 的不同数字的 3 x 3 矩阵，其中每行，每列以及两条对角线上的各数之和都相等。

给定一个由整数组成的 N × N 矩阵，其中有多少个 3 × 3 的 “幻方” 子矩阵？（每个子矩阵都是连续的）。

 

示例 1:

输入: [[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 = grid[0].length <= 10
2. 0 <= grid[i][j] <= 15

class Solution(object):
    def numMagicSquaresInside(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        res = 0
        for i in range(1,len(grid)-1):
            for j in range(1,len(grid[0])-1):
                set0 = {grid[i-1][j-1],grid[i-1][j],grid[i-1][j+1],
                        grid[i][j-1],grid[i][j],grid[i][j+1],
                       grid[i+1][j-1],grid[i+1][j],grid[i+1][j+1]}
                set1 = {1,2,3,4,5,6,7,8,9}
                S1 = grid[i-1][j]+grid[i+1][j]
                S2 = grid[i-1][j-1]+grid[i+1][j+1]
                S3 = grid[i-1][j+1]+grid[i+1][j-1]
                S4 = grid[i][j-1]+grid[i][j+1]
                if grid[i][j] == 5:
                    if (set0 == set1) and (S1 == S2 == S3 == S4 == 10):
                        res += 1
        return res