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

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

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

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

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

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

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

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

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

    return 0;
}