牛客小白月赛4C——病菌感染

链接：https://www.nowcoder.com/acm/contest/134/C 
来源：牛客网

 #include <bits/stdc++.h>
 using namespace std;
 #define N 1005
 int a[N][N],dir[][]={{-,,,},{,,,-}};//四周的格子
 int n,m,sum;

 bool judge(int x,int y)//判断周围的病菌数
 {
     int xx,yy,i;
     sum=;//周围病菌数
     for (i=;i<;i++)
     {
         xx=x+dir[][i];
         yy=y+dir[][i];
         if (xx<||xx>n||yy<||yy>n)//超出范围
         {
             continue;
         }
         if (a[xx][yy]==)
         {
             sum++;
         }
     }
     if (sum>=)
     {
         return true;
     }
     return false;
 }

 void bfs(int i,int j)//广搜标记被感染的格子
 {
     if (i<=||i>n||j<=||j>n||false==judge(i,j)||a[i][j]==)//若无a[i][j]==1则会死循环
     {
         return;
     }

     if (true==judge(i,j))
     {
         a[i][j]=;
         bfs(i,j-);
         bfs(i,j+);
         bfs(i-,j);
         bfs(i+,j);
     }
 }

 void test()
 {
     int i,j;
     for (i=;i<=n;i++)
     {
         for (j=;j<=n;j++)
         {
             printf("%d",a[i][j]);
         }
         printf("\n");
     }
 }

 void solve()
 {
     int i,j,r,l;
     memset(a,,sizeof(a));
     for (i=;i<m;i++)
     {
         scanf("%d %d",&r,&l);
         a[r][l]=;
     }
 //    test();
     for (i=;i<=n;i++)
     {
         for (j=;j<=n;j++)
         {
             if (a[i][j]==)
             {
                 bfs(i,j);
             }
         }
     }
     for (i=;i<=n;i++)//看是否全部被感染
     {
         for (j=;j<=n;j++)
         {
             if (a[i][j]==)
             {
                 printf("NO\n");
                 return;
             }
         }
      }
      printf("YES\n");
      return;
 }

 int main()
 {
     while (scanf("%d %d",&n,&m)!=EOF)
     {
         solve();
 //        test();
     }

     return ;
 }