2019暑期集训牛客多校8-A. All-one Matrices
思路
给你一个n*m由0、1组成的矩阵,要你找出所有由1组成且不被另一个更大的矩阵(也全由1组成)完全包含的矩阵。并输出个数。
最大子矩阵问题加去重,用h[i][j]存(i,j)位置上1的高度,单增栈找到每一行的每个点向左、向右所能到达的最远的位置存在L[i][j]、R[i][j]。接下来就是最大子矩阵问题了。再用单增栈去重。
题解
#include<bits/stdc++.h>
using namespace std;
const int maxn = 3005;
int h[maxn][maxn], sum[maxn][maxn], q[maxn], R[maxn], L[maxn];
char s[maxn][maxn];
int main() {
int n, m;
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) {
scanf("%s", s[i] + 1);
for (int j = 1; j <= m; j++) {
if (s[i][j] == '1')
h[i][j] = h[i - 1][j] + 1;
sum[i][j] = sum[i][j - 1] + s[i][j] - '0';
}
}
int ans = 0;
for (int i = n; i; i--) {
stack<int> sta;
sta.push(m + 1);
for (int j = m; j; j--) {
while (!sta.empty() && h[i][j] <= h[i][sta.top()])
sta.pop();
if (sta.empty())
R[j] = -1;
else
R[j] = sta.top() - 1;
sta.push(j);
}
while (!sta.empty())
sta.pop();
sta.push(0);
for (int j = 1; j <= m; j++) {
while (!sta.empty() && h[i][j] <= h[i][sta.top()])
sta.pop();
if (sta.empty())
L[j] = -1;
else
L[j] = sta.top() + 1;
sta.push(j);
}
while (!sta.empty())
sta.pop();
for (int j = 1; j <= m; j++) {
if (h[i][j] == 0) {
while (!sta.empty())
sta.pop();
continue;
}
while (!sta.empty() && h[i][j] < h[i][sta.top()])
sta.pop();
if (sta.empty() || h[i][j] != h[i][sta.top()]) {
int l = L[j];
int r = R[j];
if (sum[i +1][r] - sum[i + 1][l - 1] != r - l + 1)
ans++;
sta.push(j);
}
}
}
printf("%d\n", ans);
}