题目链接: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;
}