题意
给出一个矩形,其中 0 0 0代表完好, 1 1 1代表破损,求出从中能截出的矩形个数。
思路
容斥思想,用总的 − - −破损的方案自然得出答案。
对于一个点,我们发现只有它左边和上面的破点对它有贡献,且破点的坐标是单调递减的,即当我们的 x x x坐标越大, y y y坐标就要越小,这里使用单调栈维护距离每一列最近的破点。
代码
#include<stack>
#include<cstdio>
long long n, ans, sum;
long long a[1001][1001], last[1001][1001], f[1001][1001];
std::stack<int> s[1001];
int main() {
scanf("%d", &n);
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
scanf("%1d", &a[i][j]);
for (int i = 1; i <= n; i++)
s[i].push(0);
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++) {
sum += i * j;
if (a[i][j]) last[i][j] = j;
else last[i][j] = last[i][j - 1];
while (s[j].size() > 1 && last[s[j].top()][j] < last[i][j]) s[j].pop();
f[i][j] += f[s[j].top()][j] + (i - s[j].top()) * last[i][j];
ans += f[i][j];
s[j].push(i);
}
printf("%lld", sum - ans);
}