牛客寒假算法基础4 C

题目链接：https://ac.nowcoder.com/acm/contest/330/C

bfs基础题；

考虑到自身有两种状态，水和火，所以开三维数组，记录状态；

每一步有上下左右四种走法，不过当当前格子为道具格子时，可以选择原地不动转换状态，花费1单位时间；

需要注意的是，到一个格子的时候，状态不同是算不一样的，所以不可以通过把走过的格子变成墙来记录是否访问，可以使用flag数组记录；

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const double epos=1e-8;

struct Node{
    int x,y;
    int id;//0水;
}in,out,st,en;
const int maxn=109;
int n,m;

int step[maxn][maxn][2];//记录步数;
char mapp[maxn][maxn];
queue<Node>q;

int dx[4]={0,0,-1,1};
int dy[4]={1,-1,0,0};

int flag[maxn][maxn][2];//记录访问否;


void bfs(){
    memset(step,0,sizeof(step));
    memset(flag,0,sizeof(flag));
    q.push(st);
    flag[st.x][st.y][0]=1;
    mapp[st.x][st.y]='.';
    while(!q.empty()){
        out=q.front();
        q.pop();

        for(int i=0;i<4;i++){
            in.x=out.x+dx[i];
            in.y=out.y+dy[i];
            in.id=out.id;
            if(in.x<0||in.x>=n||in.y<0||in.y>=m) continue;
            if(mapp[in.x][in.y]=='#') continue;
            if(flag[in.x][in.y][in.id]) continue;
            if((mapp[in.x][in.y]=='~'&&in.id)||(mapp[in.x][in.y]=='w'&&in.id==0)) continue;
            step[in.x][in.y][in.id]=step[out.x][out.y][out.id]+1;
            flag[in.x][in.y][in.id]=1;
            q.push(in);
            if(mapp[in.x][in.y]=='T'){
                printf("%d\n",step[in.x][in.y][in.id]);
                return ;
            }

        }
        if(mapp[out.x][out.y]=='@'&&flag[out.x][out.y][out.id^1]==0){
                flag[out.x][out.y][out.id^1]=1;
                step[out.x][out.y][out.id^1]=step[out.x][out.y][out.id]+1;
                out.id^=1;
                q.push(out);

        }
    }
    printf("-1\n");
}


int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;++i){
        scanf("%s",mapp[i]);
        for(int j=0;j<m;++j){
            if(mapp[i][j]=='S') st.x=i,st.y=j,st.id=0;
            if(mapp[i][j]=='T') en.x=i,en.y=j,en.id=0;
        }
    }

    bfs();



    return 0;
}