【LeetCode】5454. 统计全 1 子矩形

给你一个只包含 0 和 1 的 rows * columns 矩阵 mat ，请你返回有多少个 子矩形 的元素全部都是 1 。

示例 1：

输入：mat = [[1,0,1],
[1,1,0],
[1,1,0]]
输出：13
解释：
有 6 个 1x1 的矩形。
有 2 个 1x2 的矩形。
有 3 个 2x1 的矩形。
有 1 个 2x2 的矩形。
有 1 个 3x1 的矩形。
矩形数目总共 = 6 + 2 + 3 + 1 + 1 = 13 。
示例 2：

输入：mat = [[0,1,1,0],
[0,1,1,1],
[1,1,1,0]]
输出：24
解释：
有 8 个 1x1 的子矩形。
有 5 个 1x2 的子矩形。
有 2 个 1x3 的子矩形。
有 4 个 2x1 的子矩形。
有 2 个 2x2 的子矩形。
有 2 个 3x1 的子矩形。
有 1 个 3x2 的子矩形。
矩形数目总共 = 8 + 5 + 2 + 4 + 2 + 2 + 1 = 24 。
示例 3：

输入：mat = [[1,1,1,1,1,1]]
输出：21
示例 4：

输入：mat = [[1,0,1],[0,1,0],[1,0,1]]
输出：5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-submatrices-with-all-ones
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

【题目分析】
只要枚举每一行，处理出一个left[i][j]，表示从(i,j)点开始向左能够到达的连续格子数量，然后对于每个格子，向上遍历，统计每一行的贡献即可。

0	1	1	0
0	1	1	1
1	1	1	0

例如对于上图，处理处left数组：

0	1	2	0
0	1	2	3
1	2	3	0

接下来，枚举每一个格子
例如枚举到如图的(2,2,)点

向上枚举，则根据上方已知的left不难得出每一行的贡献

就这样。。。

class Solution {
public:
    int numSubmat(vector<vector<int>>& mat) {
        int left[390][390], ans = 0, col;
        memset(left, 0, sizeof(left));
        for(int i = 0; i < mat.size(); i++){
            for(int j = 0; j < mat[i].size(); j++){
                if(!mat[i][j])left[i][j] = 0;
                else if(!j)left[i][j] = 1;
                else left[i][j] = left[i][j-1] + 1;
                col = 150;
                for(int k = i; k > -1 && left[k][j]; k--){
                    col = min(col, left[k][j]);
                    ans += col;
                }
            }
        }
            
        return ans;
    }
};