[LibreOJ β Round #4]框架

题目描述

有一个n×m的矩形框架，但其中有些边被删除了。qmqmqm想知道剩余部分中还有多少完整的正方形。只有当一个正方形的每一条边均被保留下来，这个正方形才是完整的。

做法

先预处理每个位置往四个方向的最远扩展长度。
枚举一条对角线，正方形的两个相对端点在该对角线上。
一个点作为左上角有一定修改范围，一个点作为右下角也有一定查询范围，可以用树状数组维护。

#include<cstdio>
#include<algorithm>
#include<vector>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
typedef pair<int,int> pi;
const int maxn=2000+10;
vector<pi> a[maxn];
int f[2][maxn][maxn],up[maxn][maxn],down[maxn][maxn],left[maxn][maxn],right[maxn][maxn],tree[maxn];
int i,j,k,l,t,n,m,p;
ll ans;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int lowbit(int x){
    return x&-x;
}
void change(int x){
    while (x>0){
        tree[x]++;
        x-=lowbit(x);
    }
}
int query(int x){
    int t=0;
    while (x<=p){
        t+=tree[x];
        x+=lowbit(x);
    }
    return t;
}
void solve(int xx,int yy){
    int i,j,x,y;
    vector<pi>::iterator it;
    fo(i,0,p) tree[i]=0,a[i].clear();
    i=1;
    x=xx;y=yy;
    while (x<=n&&y<=m){
        a[i-1].push_back(make_pair(i,1));
        a[i-min(down[x][y],left[x][y])-1].push_back(make_pair(i,-1));
        i++;
        x++;y++;
    }
    i=1;
    x=xx;y=yy;
    while (x<=n&&y<=m){
        change(i+min(up[x][y],right[x][y]));
        it=a[i].begin();
        while (it!=a[i].end()){
            ans+=(ll)query((*it).first)*(*it).second;
            it++;
        }
        //fo(j,0,a[i].size()-1) ans+=(ll)query(a[i][j].first)*a[i][j].second;
        i++;
        x++;y++;
    }
}
int main(){
    n=read()-1;m=read()-1;
    p=max(n,m)+1;
    fo(i,0,n)
        fo(j,0,m-1)
            f[0][i][j]=read();
    fo(i,0,n-1)
        fo(j,0,m)
            f[1][i][j]=read();
    fo(j,0,m)
        fd(i,n-1,0)
            up[i][j]=f[1][i][j]?up[i+1][j]+1:0;
    fo(j,0,m)
        fo(i,1,n)
            down[i][j]=f[1][i-1][j]?down[i-1][j]+1:0;
    fo(i,0,n)
        fo(j,1,m)
            left[i][j]=f[0][i][j-1]?left[i][j-1]+1:0;
    fo(i,0,n)
        fd(j,m-1,0)
            right[i][j]=f[0][i][j]?right[i][j+1]+1:0;
    fo(i,0,n) solve(i,0);
    fo(i,1,m) solve(0,i);
    printf("%lld\n",ans);
}