E. Zbazi in Zeydabad

树状数组的运用->好题

首先我们能想到n^3的做法，但是会超时，所以就需要优化。
我们用树状数组维护的是从这个点开始向后满足条件的情况
树状数组是维护的斜率，需要开n+m个树状数组；
用一个vector<> 去维护长度v[i]表示长度为i，且满足条件的点
我们从长度为m的开始往里面更新，那么先进去的一定是满足后面需要的条件的。

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <bitset>
#include <map>

#define FOR(i,a,b) for(int i=a;i<=b;i++)
#define ROF(i,a,b) for(int i=a;i>=b;i--)
#define mem(i,a) memset(i,a,sizeof(i))
#define rson mid+1,r,rt<<1|1
#define lson l,mid,rt<<1

#define mp make_pair
#define ll long long
#define LL long long
using namespace std;
template <typename T>inline void read(T &_x_){
    _x_=0;bool f=false;char ch=getchar();
    while (ch<'0'||ch>'9') {if (ch=='-') f=!f;ch=getchar();}
    while ('0'<=ch&&ch<='9') {_x_=_x_*10+ch-'0';ch=getchar();}
    if(f) _x_=-_x_;
}
const double eps = 0.0000001;
const int maxn = 3000+7;
const int mod = 1e9+7;

int n,m;
char s[maxn][maxn];
int l[maxn][maxn],r[maxn][maxn],lr[maxn][maxn];
vector<pair<int,int> > p[maxn];

void init(){
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            if(s[i][j]=='z') l[i][j]=l[i][j-1]+1;
            else l[i][j]=0;
        }
        for(int j=m;j>=1;j--){
            if(s[i][j]=='z') r[i][j]=r[i][j+1]+1;
            else r[i][j]=0;
        }
    }
    for(int i=n;i>=1;i--){
        for(int j=1;j<=m;j++){
            if(s[i][j]=='z'){
                lr[i][j]=lr[i+1][j-1]+1;
            }else lr[i][j]=0;
        }
    }
    for(int i=1;i<=n;i++) {  
        for(int j=1;j<=m;j++) {  
            p[j + r[i][j] - 1].push_back(mp(i,j));
        }  
    } 
}

int bit[maxn*2][maxn];
ll ans=0;  
int sum(int i, int x) {  
    int ans = 0;  
    while(x > 0) {  
        ans += bit[i][x];  
        x -= x & -x;  
    }  
    return ans;  
}  
void add(int i, int x, int d) {  
    while(x <= m) {  
        bit[i][x] += d;  
        x += x & -x;  
    }  
}

int main(){
    read(n),read(m);
    for(int i=1;i<=n;i++) scanf("%s",s[i]+1);
    init();
    for(int i=m;i>=1;i--){
        int len=p[i].size();
        for(int j=0;j<len;j++){
            int x = p[i][j].first;
            int y = p[i][j].second;
            add(x+y,y,1);
        }
        for(int j=1;j<=n;j++){
            if(s[j][i]!='z')continue;
            int len = min(l[j][i],lr[j][i]);
            ans += sum(j+i,i)-sum(i+j,i-len);
        }
    }
    printf("%I64d\n",ans);
    return 0;
}