LeetCode 1895. 最大的幻方--前缀和

1895. 最大的幻方

一个 k x k 的 幻方 指的是一个 k x k 填满整数的方格阵，且每一行、每一列以及两条对角线的和 全部相等 。幻方中的整数 不需要互不相同 。显然，每个 1 x 1 的方格都是一个幻方。

给你一个 m x n 的整数矩阵 grid ，请你返回矩阵中 最大幻方 的 尺寸 （即边长 k）。

示例 1：

输入：grid = [[7,1,4,5,6],[2,5,1,6,4],[1,5,4,3,2],[1,2,7,3,4]]
输出：3
解释：最大幻方尺寸为 3 。
每一行，每一列以及两条对角线的和都等于 12 。

• 每一行的和：5+1+6 = 5+4+3 = 2+7+3 = 12
• 每一列的和：5+5+2 = 1+4+7 = 6+3+3 = 12
• 对角线的和：5+4+3 = 6+4+2 = 12

示例 2：

输入：grid = [[5,1,3,1],[9,3,3,1],[1,3,3,8]]
输出：2

提示：

m == grid.length
n == grid[i].length
1 <= m, n <= 50
1 <= grid[i][j] <= 106

题解

和这个题目很像，LeetCode 1292. 元素和小于等于阈值的正方形的最大边长–前缀和。

AC代码

class Solution {
public:
    int sum[55][55][2];
    bool check(int ans ,int i1,int j1,int i2,int j2,vector<vector<int>>& grid)
    {
        for(int i=i1;i<=i2;i++)
        {
            if(sum[i][j2][1]-sum[i][j1-1][1]!=ans)return false;
        }
        for(int j=j1;j<=j2;j++)
        {
            if(sum[i2][j][0]-sum[i1-1][j][0]!=ans)return false;
        }
        int sum=0;
        for(int i=i2,j=j1;i>=i1;i--,j++)
        sum+=grid[i-1][j-1];
        return sum==ans?true:false;
    }
    int largestMagicSquare(vector<vector<int>>& grid) {
        memset(sum,0,sizeof(sum));
        for(int i=0;i<grid.size();i++)
        {
            for(int j=0;j<grid[i].size();j++)
            {
                sum[i+1][j+1][0]=sum[i][j+1][0]+grid[i][j];
                sum[i+1][j+1][1]=sum[i+1][j][1]+grid[i][j];
            }
        }
        int res=0;
        for(int i=1;i<=grid.size();i++)
        {
            for(int j=1;j<=grid[0].size();j++)
            {
                int ans=0;
                for(int k=0;k<100;k++)
                {
                    if(i+k>grid.size()||j+k>grid[0].size())break;
                    ans+=grid[i+k-1][j+k-1];
                    if(check(ans,i,j,i+k,j+k,grid))
                    res=max(res,k+1);
                }
            }
        }
        return res;
    }
};