LeetCode 840. 矩阵中的幻方（Javascript）

题目

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

思路

1. 因为中间的数字一定是 5，那么就从第二行的第二个数字开始右下角的倒数第二行的第二个数字找
2. 找到了 5 后，再判断矩阵的 9 个数字是否不同，如果不同的话，再看数字是否都大于 0 小于 10
3. 最后一次看中每行，每列以及两条对角线上的各数之和是否都等于 15

代码

/**
 * @param {number[][]} grid
 * @return {number}
 */
var numMagicSquaresInside = function(grid) {
    let set = new Set
    let count = 0
    for(let i = 1; i < grid.length - 1; i++) {
        let arr = grid[i]
        for(let j = 1 ; j < arr.length - 1; j++) {
            let arrBefore = grid[i - 1]
            let arrNext = grid[i + 1]
            if(arr[j] === 5 && isMagicSquares(arrBefore[j - 1], arrBefore[j], arrBefore[j + 1], arr[j - 1], arr[j], arr[j + 1], arrNext[j - 1], arrNext[j], arrNext[j + 1], set)) count++
        }
    }
    return count
};

const isMagicSquares = function(a, b, c, d, e, f, g, h, i, set) {
    set.clear()
    let isAllValid = [].every.call(arguments, (n) => {
        if(typeof n === 'number') {
            set.add(n)
        } else {
            return true
        }
        return n > 0 && n < 10
    })
    return set.size === 9 && isAllValid && a + b + c === 15 && d + e + f === 15 && g + h + i === 15 && a + d + g === 15 && c + f + i === 15
}

复杂度分析

• 时间复杂度$O(RC)$
  R 为矩阵的行长度， C 为矩阵列长度
• 空间复杂度$O(1)$