LeetCode：1292. 元素和小于等于阈值的正方形的最大边长（二维前缀和）

1292. 元素和小于等于阈值的正方形的最大边长

给你一个大小为 m x n 的矩阵 mat 和一个整数阈值 threshold。

请你返回元素总和小于或等于阈值的正方形区域的最大边长；如果没有这样的正方形区域，则返回 0 。

示例 1：

输入：mat = [[1,1,3,2,4,3,2],[1,1,3,2,4,3,2],[1,1,3,2,4,3,2]], threshold = 4
输出：2
解释：总和小于 4 的正方形的最大边长为 2，如图所示。
示例 2：

输入：mat = [[2,2,2,2,2],[2,2,2,2,2],[2,2,2,2,2],[2,2,2,2,2],[2,2,2,2,2]], threshold = 1
输出：0
示例 3：

输入：mat = [[1,1,1,1],[1,0,0,0],[1,0,0,0],[1,0,0,0]], threshold = 6
输出：3
示例 4：

输入：mat = [[18,70],[61,1],[25,85],[14,40],[11,96],[97,96],[63,45]], threshold = 40184
输出：2

需要储备二维前缀和知识点，我也是刚刚才学，建议拿个笔记本，一点一点的推演，明白了原理，倒也不是很难，毕竟直角坐标系也比较容易理解，不过第一次接触这种题，难点是理解题目，矩阵确实不好理解

关于二维前缀的知识点，建议看LeetCode的题，这里贴下链接
https://leetcode-cn.com/problems/maximum-side-length-of-a-square-with-sum-less-than-or-equal-to-threshold/

class Solution {
public:
        int get(const vector<vector<int>>& pre, int m, int n, int x, int y) {
        x = max(min(x, m), 0);
        y = max(min(y, n), 0);
        return pre[x][y];
    }
    vector<vector<int>> matrixBlockSum(vector<vector<int>>& mat, int K) {
           int m = mat.size(), n = mat[0].size();
        vector<vector<int>> p(m + 1, vector<int>(n + 1));
        for (int i = 1; i <= m; ++i) {
            for (int j = 1; j <= n; ++j) {
                p[i][j] = p[i - 1][j] + p[i][j - 1] - p[i - 1][j - 1] + mat[i - 1][j - 1];
            }
        }
        vector<vector<int>>ans(m,vector<int>(n));
        for (int i = 0; i < m; ++i) {
            for (int j = 0; j < n; ++j) {
                ans[i][j] = get(p, m, n, i + K + 1, j + K + 1) - get(p, m, n, i - K, j + K + 1) - get(p, m, n, i + K + 1, j - K) + ge(p, m, n, i - K, j - K);
            }
        }
    return ans;

    }
};