duxing的奇妙冒险2

duxing的奇妙冒险2

Description

duxing201606来到了一个迷宫！迷宫是一个n×m的矩阵，矩阵内的每个格子内可能没有东西，也可能有一个女生，也可能是有一个障碍物。duxing201606现在位于迷宫的起点s。现在duxing201606想要在迷宫内找到2个女生，最后再达到迷宫的终点t。

duxing201606每秒钟可以从当前的位置移动到上下左右四个格子中的任意一个，只要满足移动后仍在迷宫的内部，并且移动到的位置上没有障碍物。

Input

输入一行两个数n，m表示迷宫大小，接下来是n<=100, m<=1000的矩阵s表示起点，t表示终点，*表示障碍物，.表示空通路，M表示有女生。保证只有一个起点一个终点。

Output

duxing201606从起点出发见到两个妹纸最后到达终点的最少时间。如果找不到就输出-1。

Sample Input

4 5
sMMMM
MMMMM
MMMMM
MMMMt

Sample Output

7

#include<bits/stdc++.h>

using namespace std;
char s[107][1007];
int vis[107][1007];
int Road[107][1007][2];
int num[107][1007];
int dir[4][2]= {{1,0},{-1,0},{0,-1},{0,1}};
struct Node
{
    int female,x,y,cnt,nid;//female记录找到的女生数，nid与num数组对应，记录找到的女生的序号
};
int n,m,cnt,newx,newy;
int Sx,Sy,Tx,Ty;
queue <Node> q;

int dfs()
{
    q.push({0,Sx,Sy,0,0});
    vis[Sx][Sy]=1;//起点墙
    while(!q.empty())
    {
        Node tmp=q.front();
        q.pop();
        if(tmp.female==2&&tmp.x==Tx&&tmp.y==Ty)//终点
        {
            return tmp.cnt;
        }
        else if(tmp.female==0)//找第一个女生
        {
            for(int i=0; i<4; i++)
            {
                newx=tmp.x+dir[i][0];
                newy=tmp.y+dir[i][1];
                if(newx>=1&&newx<=n&&newy>=1&&newy<=m&&s[newx][newy]!='*'&&vis[newx][newy]==0)//非墙&&没找过该位置
                //注意第二次找后的返回后vis变化仍可能进入此循环，
                //不可写成vis!=1
                {
                    Road[newx][newy][0]=num[newx][newy];//female==0时，记录该处女生序号
                    q.push({(s[newx][newy]=='M'),newx,newy,tmp.cnt+1,num[newx][newy]});//记录女生序号，是女生即返回1
                    vis[newx][newy]=1;
                }
            }
        }
        else if(tmp.female==1)//找第二个女生
        {
            for(int i=0; i<4; i++)
            {
                newx=tmp.x+dir[i][0];
                newy=tmp.y+dir[i][1];
                if(newx>=1&&newx<=n&&newy>=1&&newy<=m&&s[newx][newy]!='*'&&Road[newx][newy][1]==0&&Road[newx][newy][0]!=tmp.nid&&Road[newx][newy][1]!=tmp.nid)
                //非墙&&female==1时，没找过此处&&female==0时，没找过此处&&第二次和第一次找的不是一处
                //此处理解为用这三种判断取代vis作用，避免重复计算找第一次找过的地点及避免找第二次找过处
                {
                    if(Road[newx][newy][0])//第一次还没找过此处
                        Road[newx][newy][1]=tmp.nid;//变墙
                    else//第一次找过此处
                        Road[newx][newy][0]=tmp.nid;//变墙
                    q.push({1+(s[newx][newy]=='M'),newx,newy,tmp.cnt+1,tmp.nid});//继续保存第一个女生序号，是女生返回2
                }

            }
        }
        else if(tmp.female==2)//找起点处
        {
            for(int i=0; i<4; i++)
            {
                newx=tmp.x+dir[i][0];
                newy=tmp.y+dir[i][1];
                if(newx>=1&&newx<=n&&newy>=1&&newy<=m&&s[newx][newy]!='*'&&vis[newx][newy]!=2)//非墙&&找起点时没找过
                {
                    q.push({2,newx,newy,tmp.cnt+1,0});
                    vis[newx][newy]=2;
                }

            }
        }

    }
    return -1;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++)
        scanf("%s",s[i]+1);
    for(int i=1; i<=n; i++)
        for(int j=1; j<=m; j++)
        {
            if(s[i][j]=='M')
            {
                num[i][j]=++cnt;
            }
            if(s[i][j]=='s')
            {
                Sx=i;
                Sy=j;
            }
            if(s[i][j]=='t')
            {
                Tx=i;
                Ty=j;
            }

        }
    printf("%d",dfs());
    return 0;
}