BZOJ 1201 数三角形

暴力
枚举底边
预处理左右延伸长度
注意标号
蒟蒻我分不清l,r

#include <cstdio>

const int MAXN=1011;

int N;
int Map[3][MAXN][MAXN];
int Cnt[3][2][MAXN][MAXN];

int main(){
    
    scanf("%d", &N);
    
    for(int i=1;i<=N;++i){
        for(int j=1, a, b, c;j<=i;++j){
            scanf("%d%d%d", &a, &b, &c);
            Map[0][i][j]=c;
            Map[2][N-j+1][i-j+1]=a;
            Map[1][N-i+j][N-i+1]=b;
        }
    }
    
    for(int k=0, c;k<3;++k){
        for(int i=1;i<=N;++i){
            c=0;
            for(int j=1;j<=i;++j){
                if(!Map[k][i][j])   c=0;
                else    ++c;
                Cnt[k][0][i][j]=c;
            }
        }
    }
    
    for(int k=0, c;k<3;++k){
        for(int i=1;i<=N;++i){
            c=0;
            for(int j=i;j>=1;--j){
                if(!Map[k][i][j])   c=0;
                else    ++c;
                Cnt[k][1][i][j]=c;
            }
        }
    }
    
    /*
    for(int k=0;k<3;++k){
        for(int i=1;i<=N;++i){
            for(int j=1;j<=i;++j){
                printf("%d ", Cnt[k][1][i][j]);
            }
            puts("");
        }
        puts("");
    }
    */
    
    long long ANS=0LL;
    for(int i=1, d;i<=N;++i){
        for(int l=0;l<i;++l){
            for(int r=l+1;r<=i;++r){
                d=r-l;
                if(Cnt[0][0][i][r]<d)   continue;
                if(Cnt[1][1][N-i+r][N-i+1]>=d && Cnt[2][0][N-l][i-l]>=d){
                    ANS+=1LL;
                    //printf("%d %d %d u\n", i, l, r);
                }
                if(Cnt[2][1][N-r][i-r+1]>=d && Cnt[1][0][N-i+l][N-i]>=d){
                    ANS+=1LL;
                    //printf("%d %d %d d\n", i, l, r);
                }
            }
        }
    }
    
    printf("%lld\n", ANS);
    
    return 0;
}

/*
5
1 1 1
1 1 0 1 1 0
1 1 1 1 1 1 1 0 1
1 0 1 1 1 1 0 1 1 1 1 1
0 1 1 1 1 1 0 1 1 1 1 1 0 1 1

19

*/