Leetcode.1504 统计全 1 子矩形

感觉画质不如…原神

于 2023-05-01 20:42:08 发布

阅读量91

点赞数

分类专栏： Leetcode 文章标签：算法 leetcode 动态规划

本文链接：https://blog.csdn.net/m0_74396439/article/details/130458495

版权

Leetcode 专栏收录该内容

441 篇文章 2 订阅

订阅专栏

题目链接

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

题目描述

给你一个 m x n 的二进制矩阵 $ma t$ ，请你返回有多少个 子矩形 的元素全部都是 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$
$ma t [i] [j]$ 仅包含 0 或 1

解法：动态规划

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

在这里插入图片描述

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

在这里插入图片描述
以 1 为右下角的子矩阵为： $[1], [1, 2], [1, 4], [1, 2, 3, 4], [1, 4, 5], [1, 4, 5, 6]$ 。

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

即：

$row(i,j)=\left\{ \begin{aligned} row(i,j) & = 0 , mat[i][j] = 0\\ row(i,j) & = row(i,j - 1) + 1,mat[i][j] = 1 \\ \end{aligned} \right.$

那么以 $ma t [i] [j]$ 为右下角的子矩阵数量为 $\sum_{k = i}^{0}(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;
    }
};