CodeForces - 11C How Many Squares?【DFS】

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

来自洛谷题解的做法，先找第一类正方形，再找第二类正方形，dfs找联通块并全部清除，判断1的数量是否构成之前找到的任意一种正方形。
官方给出的2200分标的有点过高了

#include <iostream>
#include <cstring>
using namespace std;
static const int MAXN=250+10;
static const int dx[]={-1,-1,-1,0,0,1,1,1};
static const int dy[]={-1,0,1,-1,1,-1,0,1};
char g[MAXN][MAXN];
int T,n,m;
int check1(int x,int y)
{
    int a=0,b=0,c=0,d=0;
    while(x+1<n && g[x+1][y]=='1') a++,x++;
    while(y+1<m && g[x][y+1]=='1') b++,y++;
    while(x-1>=0 && g[x-1][y]=='1') c++,x--;
    while(y-1>=0 && g[x][y-1]=='1') d++,y--;
    if(a==b && b==c && c==d) return a;
    return 0;
}
int check2(int x,int y)
{
    int a=0,b=0,c=0,d=0;
    while(x+1<n && y-1>=0 && g[x+1][y-1]=='1') x++,y--,a++;
    while(x+1<n && y+1<m && g[x+1][y+1]=='1') x++,y++,b++;
    while(x-1>=0 && y+1<m && g[x-1][y+1]=='1') x--,y++,c++;
    while(x-1>=0 && y-1>=0 && g[x-1][y-1]=='1') x--,y--,d++;
    if(a==b && b==c && c==d) return a;
    return 0;
}
int num;
void check3(int x,int y)
{
    g[x][y]='0'; num++;
    for(int i=0;i<8;i++)
    {
        int nx=x+dx[i],ny=y+dy[i];
        if(0<=nx && nx<n && 0<=ny && ny<m && g[nx][ny]=='1') check3(nx,ny);
    }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
            scanf("%s",g[i]);
        int res=0;
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
                if(g[i][j]=='1')
                {
                    int a=check1(i,j),b=check2(i,j);
                    num=0; check3(i,j);
                    if(num==4*a || num==4*b) res++;
                }
        printf("%d\n",res);
    }
    return 0;
}