1504 统计全 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

提示:
1 <= rows <= 150
1 <= columns <= 150
0 <= mat[i][j] <= 1

方法1:
主要思路:解题汇总链接
(1)使用二维数组 left_ones[ i ][ j ]辅助计算,表示原数组mat在位置 (i,j)开始,在第 i 行,从第 j列向前,有多少个连续的1;
(2)当再次遍历数组mat时,对于新的 (i,j)位置,新增的矩形数量可以是以(i,j)为矩形右下角的新的矩形的数量,则可以通过 left_ones[ k ][ j ] ,其中 k>=0&&k<=i,来计算可以新增的矩形的数量,进行统计;

class Solution {
public:
    int numSubmat(vector<vector<int>>& mat) {
        int rows=mat.size();
        int cols=mat[0].size();
        vector<vector<int>> left_ones(rows,vector<int>(cols,0));//统计原数组位置(i,j),在第 i 行,从第 j 列开始,向左连续的1的个数;
        for(int i=0;i<rows;++i){
            int cur=0;
            for(int j=0;j<cols;++j){
                if(mat[i][j]==0){
                    cur=0;
                }
                else{
                    ++cur;
                }
                left_ones[i][j]=cur;
            }
        }
        int res=0;//存储结果
        for(int i=0;i<rows;++i){
            for(int j=0;j<cols;++j){
                int min_ones=INT_MAX;
                for(int k=i;k>=0;--k){//以当前的left_ones[ i ][ j ]为低,可以形成的新增的矩形的数量
                    min_ones=min(min_ones,left_ones[k][j]);
                    if(min_ones==0){//说明再向上也不能形成连续的矩形了,故可以终止统计;
                        break;
                    }
                    res+=min_ones;
                }
            }
        }
        return res;
    }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值