LeetCode1504. Count Submatrices With All Ones

文章目录

一、题目

Given an m x n binary matrix mat, return the number of submatrices that have all ones.

Example 1:

Input: mat = [[1,0,1],[1,1,0],[1,1,0]]

Output: 13

Explanation:

There are 6 rectangles of side 1x1.

There are 2 rectangles of side 1x2.

There are 3 rectangles of side 2x1.

There is 1 rectangle of side 2x2.

There is 1 rectangle of side 3x1.

Total number of rectangles = 6 + 2 + 3 + 1 + 1 = 13.

Example 2:

Input: mat = [[0,1,1,0],[0,1,1,1],[1,1,1,0]]

Output: 24

Explanation:

There are 8 rectangles of side 1x1.

There are 5 rectangles of side 1x2.

There are 2 rectangles of side 1x3.

There are 4 rectangles of side 2x1.

There are 2 rectangles of side 2x2.

There are 2 rectangles of side 3x1.

There is 1 rectangle of side 3x2.

Total number of rectangles = 8 + 5 + 2 + 4 + 2 + 2 + 1 = 24.

Constraints:

1 <= m, n <= 150

mat[i][j] is either 0 or 1.

二、题解

cpp 复制代码
class Solution {
public:
    int numSubmat(vector<vector<int>>& mat) {
        int res = 0;
        int m = mat.size(),n = mat[0].size();
        vector<int> heights(n,0);
        for(int i = 0;i < m;i++){
            for(int j = 0;j < n;j++){
                heights[j] = mat[i][j] == 0 ? 0 : heights[j] + 1;
            }
            res += countFromBottom(heights);
        }
        return res;
    }
    int countFromBottom(vector<int>& heights){
        int n = heights.size();
        int res = 0;
        stack<int> st;
        for(int i = 0;i < n;i++){
            while(!st.empty() && heights[i] <= heights[st.top()]){
                int cur = st.top();
                st.pop();
                if(heights[cur] > heights[i]){
                    int left = st.empty() ? -1 : st.top();
                    int len = i - left - 1;
                    int bottom = max(left == -1 ? 0 : heights[left],heights[i]);
                    res += (heights[cur] - bottom) * len * (len + 1) / 2;
                }
            }
            st.push(i);
        }
        while(!st.empty()){
            int cur = st.top();
            st.pop();
            int left = st.empty() ? -1 : st.top();
            int len = n - left - 1;
            int bottom = left == -1 ? 0 : heights[left];
            res += (heights[cur] - bottom) * len * (len + 1) / 2;
        }
        return res;
    }
};
相关推荐
感哥4 小时前
C++ 多态
c++
沐怡旸11 小时前
【底层机制】std::string 解决的痛点?是什么?怎么实现的?怎么正确用?
c++·面试
NAGNIP12 小时前
大模型框架性能优化策略:延迟、吞吐量与成本权衡
算法
美团技术团队13 小时前
LongCat-Flash:如何使用 SGLang 部署美团 Agentic 模型
人工智能·算法
River41614 小时前
Javer 学 c++(十三):引用篇
c++·后端
感哥17 小时前
C++ std::set
c++
Fanxt_Ja17 小时前
【LeetCode】算法详解#15 ---环形链表II
数据结构·算法·leetcode·链表
侃侃_天下17 小时前
最终的信号类
开发语言·c++·算法
茉莉玫瑰花茶17 小时前
算法 --- 字符串
算法
博笙困了18 小时前
AcWing学习——差分
c++·算法