Leetcode.1504 统计全 1 子矩形

题目链接

Leetcode.1504 统计全 1 子矩形 Rating ： 1845

题目描述

给你一个 m x n 的二进制矩阵 $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 。

提示：

• $1<=m,n<=150$
• $mat[i][j]$ 仅包含 0 或 1

解法：动态规划

如果 $mat[i][j]=1$的话，我们就记录 以 $(i,j)$ 为右下角的子矩阵有多少个。 这样统计的话，答案一定是不重不漏地。

答案是 6 个。将其看作右边那张图。

以 1 为右下角的子矩阵为 ：$[1],[1,2],[1,4],[1,2,3,4],[1,4,5],[1,4,5,6]$。

我们定义 $row(i,j)$ 为从 $mat[i][j]$ 开始，延申向左边的连续的 1 的个数。

即：

$$row(i,j)=\{\begin{array}{rl}row(i,j)& =0,mat[i][j]=0\\ row(i,j)& =row(i,j-1)+1,mat[i][j]=1\end{array}$$

那么以 $mat[i][j]$ 为右下角的子矩阵数量为 ∑ k = i 0 ( m i n { r o w ( k , j ) } ) \sum_{k = i}^{0}(min\{ row(k,j) \}) ∑k=i0​(min{row(k,j)})。

时间复杂度： $O(m^2\times n)$

C++代码：

class Solution {
public:
    int numSubmat(vector<vector<int>>& mat) {
        int m = mat.size() , n = mat[0].size();
        vector<vector<int>> row(m,vector<int>(n));
        for(int i = 0;i < m;i++){
            for(int j = 0;j < n;j++){
                if(j == 0){
                    row[i][j] = mat[i][j];
                    continue;
                }
                if(mat[i][j]) row[i][j] = row[i][j-1] + 1;
                else row[i][j] = 0;
            }
        }

        int ans = 0;
        for(int i = 0;i < m;i++){
            for(int j = 0;j < n;j++){
                int t = row[i][j];
                for(int k = i;k >= 0 && t;k--){
                    t = min(t , row[k][j]);
                    ans += t;
                }
            }
        }
        return ans;
    }
};