题解 | #数芝麻#

dfs题解

using namespace std;
int a[105][105];
int vis[105][105];
int dx[]={0,1,0,-1};
int dy[]={1,0,-1,0};
int n,m,flag;
int ans;
void dfs(int x,int y)
{
    vis[x][y]=1;
    for(int i=0;i<4;i++)
    {
        int tx=x+dx[i];
        int ty=y+dy[i];
        if(tx>=1&&tx<=n&&ty>=1&&ty<=n&&vis[tx][ty]==0&&a[tx][ty]==a[x][y])
        {
            ans+=a[tx][ty];
            dfs(tx,ty);
        }
    }
}
int main()
{
	while(cin>>n>>m)
    {
        if(n==0&&m==0)
        {
            return 0;
        }
        memset(vis,0,sizeof(vis));
        memset(a,0,sizeof(a));
        flag=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                cin>>a[i][j];
            }
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                ans=a[i][j];

                if(a[i][j]!=0&&vis[i][j]==0)
                {
                    dfs(i,j);
                    if(ans==m)
                    {
                        flag=1;
                    }
                }
            }
        }
        if(flag==1)
        {
            cout<<"YES"<<endl;
        }else
        {
            cout<<"NO"<<endl;
        }

    }
    return 0;
}