clean magic matrix code example

Example: clean magic matrix

int largestMagicSquare(vector<vector<int>>& grid) {
    int m = grid.size(), n = grid[0].size(), res = 0;
    vector<vector<int>> rows(m + 2, vector<int>(n + 2)), cols(rows), d1(rows), d2(rows);
    for (int i = 1; i <= m; ++i)
        for (int j = 1; j <= n; ++j) {
            rows[i][j] += grid[i - 1][j - 1] + rows[i][j - 1];
            cols[i][j] += grid[i - 1][j - 1] + cols[i - 1][j];
            d1[i][j] += grid[i - 1][j - 1] + d1[i - 1][j - 1];
            d2[i][j] += grid[i - 1][j - 1] + d2[i - 1][j + 1];
        }
    for (int i = 1; i <= m; ++i)
        for (int j = 1; j <= n; ++j)
            for (int k = min(m - i, n - j); k > res; --k) {
                int sum_d1 = d1[i + k][j + k] - d1[i - 1][j - 1];
                int sum_d2 = d2[i + k][j] - d2[i - 1][j + k + 1];
                bool match = sum_d1 == sum_d2; 
                for (int l = 0; l <= k && match; ++l) {
                    match &= sum_d1 == rows[i + l][j + k] - rows[i + l][j - 1];
                    match &= sum_d1 == cols[i + k][j + l] - cols[i - 1][j + l];
                }
                if (match)
                    res = k;
            }
    return res + 1;
}

Tags:

Misc Example