题目描述:
给定一个二维数组matrix,其中的值不是0就是1,返回全部由1组成的子矩阵的数量。
way:
假设我们遍历矩形的每一行,以当前遍历到的行作为地基,去看这一行的直方图(直方图介绍 ->直方图的最大长方形面积-CSDN博客)中全部由1组成的子矩阵的个数,将每一行得到的子矩阵数加起来,得到答案,将二维问题转变为了一维问题,所以就是像下面这样。
第一行:1 1 1 1 0 1
第二行:2 2 2 2 0 2
第三行:3 3 3 3 1 3
第四行: 0 4 4 4 2 4 (做地基时如果有0直接为0,地基都没有,上面的没法垒起来的)
第五行:1 5 0 5 3 5
为了明显点,假设这5行中的有一行是 3, 2, 4, 2, 5,那么计算这一行的直方图的子矩阵的数量就是,以当前行的每个元素作为所在子矩阵中的最小高度,得到一块区域,假设以第4个2作为最低高度,那么可以得到如下的一块区域,在这个区域中,对于高度为2的子矩阵的数量,有1-1列,1-2列,1-3列,1-4列,1-5列,2-2列,2-3,2-4,2-5,3-3,3-4,3-5,4-4,4-5,5-5 这(1+5)*5/2=15个,对于高度为1的数量,也有15个,所以如果区域长度为L,高度为h,那么子矩阵的数量就是 L * (L+1)/2 * h 个,对于上面说的,有 5 *(5+1)/2 *2 = 30个,这个高度h为当前选的最低高度 - max(left, right) (left,right 分别为它的左右两边离它最近且比它 小于or等于 的数,就像第一个2,左边比它小的数没有,右边比它 等于 的数是第2个2,然后h=2-2=0, 没有计算以第一个2为最小高度时的子矩阵数,到了第2个2的时候,左边比它小的数没有,右边比它小的数也没有,h=2-0=2, 到了第2个2的时候才计算了下面的阴影部分的子矩阵的数量,这样就不会算重复,只会在最后一个2的时候去计算以2为最低高度的时候那块高度为2连通的大区域的子矩阵数,必须是 小于等于,必须是 h = 高度差,这样就不会在 最低高度 =3 的时候去计算高度为2的子矩阵数,那样高度为2的区域其实还没有完全连起来形成一块大区域,会算重)。注意要一行一行的做地基 (地基必被算)得到一个个直方图数组去算去加和,其实你按我上面的说法1-1,1-2,1-3....,不同高度这样一行一行的去算就知道对不对了。
#include<iostream>
#include<vector>
#include<stack>
using namespace std;int subMatrixCount(vector<int> height)
{int N = height.size();stack<int>st;int num = 0;for (int i = 0; i < N; i++){while (!st.empty() && height[i] <= height[st.top()]){int topIndex = st.top();st.pop();int leftIndex = st.empty() ? -1 : st.top();int rightIndex = i;//(rightIndex-1)-(leftIndex+1) + 1int L = rightIndex - 1 - leftIndex;int h = height[topIndex] - max(height[rightIndex], leftIndex >= 0 ? height[leftIndex] : 0);num += L * (L + 1) / 2 * h;}st.push(i);}while (!st.empty()){int topIndex = st.top();st.pop();int leftIndex = st.empty() ? -1 : st.top();int rightIndex = N;//(rightIndex-1)-(leftIndex+1) + 1int L = rightIndex - 1 - leftIndex;int h = height[topIndex] - max(0, leftIndex >= 0 ? height[leftIndex] : 0);num += L * (L + 1) / 2 * h;}return num;
}int allSubMatrixCountBy1(vector<vector<char>> mp)
{if (mp.size() == 0) return 0;int M = mp.size();int N = mp[0].size();vector<int>row(N);int sum = 0;for (int i = 0; i < M; i++){for (int j = 0; j < N; j++){if (mp[i][j] == '1'){row[j] += 1;}else{row[j] = 0;}}sum += subMatrixCount(row);}return sum;
}int main()
{vector<vector<char>> mp = {{'1', '1'},{'1', '1'}};int sum = allSubMatrixCountBy1(mp);cout << sum << endl; //9return 0;
}