Codeforces Round #567 (Div. 2)C. Flag(暴力)

题目链接：https://codeforces.com/contest/1181/problem/C

 

题目大意：找出符合条件的矩形个数，条件是一个矩形分为上中下三层，每层颜色一样且长度一样

 

题目思路：先预处理出每个块即块上面一共有多少个与之相连且颜色相同的块，然后枚举每一块，假设该块作为矩形的最下面那一块是否符合要求。符合的条件是要求a[i-k][j]也必须是k(中间那一块的长度相同)，a[i-2*k][j]需要大于等于k(大于可以只截取等于k的部分)。除此之外还要考虑合成大矩形的情况。n个可以组合的小矩形可以拼成(1+n)*n/2种情况的矩形，所以可以记录一下每个小矩形的高度，然后对于下一个是否能跟上一个合并的条件一个是是否高度相同，然后就是判断是否每层的颜色都相同。

具体实现可以参照代码，思路看着有点迷糊的可以直接看代码，可以直接看懂。

 

以下是代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<math.h>
#include<vector>
#include<stack>
using namespace std;
#define inf 0x3f3f3f3f
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
const int MAXN = 1e3+5;
const int MOD = 1e9+7;
char mp[MAXN][MAXN];
int n,m,a[MAXN][MAXN],f[MAXN][MAXN];
int main(){
    while(~scanf("%d%d",&n,&m)){
        memset(f,0,sizeof(f));
        int ans=0;
        rep(i,1,n){
            scanf("%s",mp[i]+1);
        }
        rep(i,1,n){
            rep(j,1,m){
                if(i==1||mp[i][j]!=mp[i-1][j]){
                    a[i][j]=1;
                }
                else{
                    a[i][j]=a[i-1][j]+1;
                }
            }
        }
        rep(i,3,n){
            int l=0;
            rep(j,1,m){
                int k=a[i][j];
                if(a[i-k][j]!=k||a[i-2*k][j]<k){
                    l=0;
                    continue;
                }
                f[i][j]=1;
                if(l==k&&mp[i][j]==mp[i][j-1]&&mp[i-k][j]==mp[i-k][j-1]&&mp[i-2*k][j]==mp[i-2*k][j-1]){
                    f[i][j]+=f[i][j-1];
                }
                l=k;
                ans+=f[i][j];
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}