洛谷 P1363 dfs

https://www.luogu.org/problemnew/show/P1363

题目描述

背景 Background

（喵星人LHX和WD同心协力击退了汪星人的入侵，不幸的是，汪星人撤退之前给它们制造了一片幻象迷宫。）

WD：呜呜，肿么办啊……

LHX：momo...我们一定能走出去的！

WD：嗯，+U+U！

描述 Description

幻象迷宫可以认为是无限大的，不过它由若干个N*M的矩阵重复组成。矩阵中有的地方是道路，用'.'表示；有的地方是墙，用'#'表示。LHX和WD所在的位置用'S'表示。也就是对于迷宫中的一个点(x,y)，如果(x mod n,y mod m)是'.'或者'S'，那么这个地方是道路；如果(x mod n,y mod m)是'#'，那么这个地方是墙。LHX和WD可以向上下左右四个方向移动，当然不能移动到墙上。

请你告诉LHX和WD，它们能否走出幻象迷宫（如果它们能走到距离起点无限远处，就认为能走出去）。如果不能的话，LHX就只好启动城堡的毁灭程序了……当然不到万不得已，他不想这么做。。。

输入输出格式

输入格式：

 

输入格式 InputFormat

输入包含多组数据，以EOF结尾。

每组数据的第一行是两个整数N、M。

接下来是一个N*M的字符矩阵，表示迷宫里(0,0)到(n-1,m-1)这个矩阵单元。

 

输出格式：

 

输出格式 OutputFormat

对于每组数据，输出一个字符串，Yes或者No。

 

输入输出样例

输入样例#1： 复制

5 4
##.#
##S#
#..#
#.##
#..#
5 4
##.#
##S#
#..#
..#.
#.##

输出样例#1： 复制

Yes
No

说明

数据范围和注释 Hint

对于30%的数据，N,M<=20

对于50%的数据，N.M<=100.

对于100%的数据，N,M<=1500，每个测试点不超过10组数据.

思路：这题还挺有意思的。能走到起点无限远就意味着可以从起始点沿着某个方向一直走，而不会走回头路回到起点。(进入循环)因此我们dfs要用到4个参数：x，y，fx，fy，其中x、y代表猫在地图上的位置，也即当x<0||x>=n||y<0||y>=m的时候是要对x和y进行操作的，但是fx和fy代表当前位置相对初始位置的移动，无需做额外操作，那么当fx==x&&fy==y且该点被访问过的时候，说明这是一个死循环，并不是解。因为沿着某个方向若能回到起点， 必定是满足fx!=x||fy!=y的。 那么我们考虑用一个三维vis数组，每次dfs令：vis[x][y][1]=fx，vis[x][y][2]=fy，vis[x][y][0]=1，那么当vis[x][y][0]&&(vis[x][y][1]!=x||vis[x][y][2]!=y)的时候，说明我们找到了一个满足题意的解。即经过有效的移动又走到了以前走过的点，那么必定可以沿着这条路径一直走下去。若vis[x][y][0]=1但是不满足上式，说明走了回头路，筛去即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#include<algorithm>
using namespace std;

int vis[1505][1505][3];
char s[2000][2000];
int n,m;
int flag=0;
int d[4][2]={{-1,0},{1,0},{0,-1},{0,1}};

void dfs(int x,int y,int fx,int fy)
{
    if(flag)
        return;
    if(vis[x][y][0]&&(vis[x][y][1]!=fx||vis[x][y][2]!=fy))//重复走过的点
    {
        flag=1;
        return;
    }
    else if(vis[x][y][0])
        return ;
    vis[x][y][0]=1;
    vis[x][y][1]=fx;
    vis[x][y][2]=fy;
    for(int i=0;i<4;i++)
    {
        int dx=x+d[i][0];
        int dy=y+d[i][1];
        dx=(dx+n)%n;
        dy=(dy+m)%m;
        if(s[dx][dy]!='#')
            dfs(dx,dy,fx+d[i][0],fy+d[i][1]);
    }
}

int main()
{
    while(~scanf("%d %d",&n,&m))
    {
        flag=0;
        memset(vis,0,sizeof(vis));
        int x=0,y=0;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                cin>>s[i][j];
                if(s[i][j]=='S')
                    x=i,y=j;
            }
        }
        if(n==1&&m==1)
        {
            printf("Yes\n");
            continue;
        }
        dfs(x,y,x,y);
        if(flag)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}