题意:给出一个由字符‘z'和’.'组成的图,问这个图中有多少个正方形,满足上边的边和下面的边以及向左的对角线上全是z的情形(即正方形内有一个最大的Z)。
思路:暴力的做法是预处理出每个点向左到达的最远距离和反对角线所能到达的最远距离,然后枚举z右上角的那个点,根据预处理出的两个数值算出可能的最远的左下角,然后对于右上角和左下角之间的点一个个验证他们最右能到达的距离,这样做的复杂度会到达O(n^3)无法承受。
注意到上面一步中有一个处理区间的问题,所以可以考虑用树状数组优化,只要我们记录对于当前右上角所在对角线上哪些点满足条件即可,然后用树状数组求得区间内的和。
可以发现,区间内的一个点n被标记当且仅当在右上角右边的列(包括右上角所在列)对应n的那一行的点能到达当前对角线,所以我们可以这样做:
从右到左遍历每一列,对于一个点如果它是一条线段的右端点,那么对于这条线段上所有向左的点,标记那个点所在对角线上的当前行,然后在枚举这一列上所有点为z右上角点,然后用树状数组求和即可,因为是从右向左开始更新树状数组的,所以可以保证答案的正确性。
#include<bits/stdc++.h>
#define eps 1e-6
#define LL long long
#define pii pair<int, int>
#define pb push_back
#define mp make_pair
//#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
const int MAXN = 3100;
//const int INF = 0x3f3f3f3f;
int n, m;
LL c[MAXN<<1][MAXN];
int l[MAXN][MAXN], dia[MAXN][MAXN];
char G[MAXN][MAXN];
bool flag[MAXN][MAXN];
int lowbit(int x)
{
return x&-x;
}
void add(int id, int pos)
{
while (pos <= n) {
c[id][pos]++;
pos += lowbit(pos);
}
}
LL sumv(int id, int pos)
{
int ret = 0;
while (pos > 0) {
ret += c[id][pos];
pos -= lowbit(pos);
}
return ret;
}
void init()
{
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= m; j++) {
if (G[i][j] != 'z')
l[i][j] = 0;
else {
l[i][j] = l[i][j-1] + 1;
if (G[i][j+1] != 'z')
flag[i][j] = true;
}
}
}
for (int i = n; i > 0; i--) {
for (int j = 1; j <= m; j++) {
if (G[i][j] != 'z')
dia[i][j] = 0;
else {
dia[i][j] = dia[i+1][j-1] + 1;
}
}
}
}
int main()
{
//freopen("input.txt", "r", stdin);
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++)
scanf("%s", G[i]+1);
init();
LL ans = 0;
for (int j = m; j > 0; j--) {
for (int i = 1; i <= n; i++)
if (flag[i][j]) {
int pos = j;
while (G[i][pos] == 'z') {
add(i+pos, i);
pos--;
}
}
for (int i = 1; i <= n; i++) {
if (G[i][j] == 'z')
ans += sumv(i+j, i+min(l[i][j], dia[i][j])-1) - sumv(i+j, i-1);
}
}
cout << ans;
return 0;
}