HDU 1175 连连看【BFS】

题意：给出起点和终点的棋子，不能经过别的棋子，而且转弯的次数不能超过2次，问能否消除

和逃离迷宫一样，每个节点记录下来它的当前的方向和转弯的次数，再搜

注意特判起点的棋子和终点的棋子为0或者不一样的情况，这样的话就不用搜了，直接输出NO

#include<iostream>  
#include<cstdio>  
#include<cstring> 
#include <cmath> 
#include<stack>
#include<vector>
#include<map> 
#include<set>
#include<queue> 
#include<algorithm>  
using namespace std;

typedef long long LL;
const int INF = (1<<30)-1;
const int mod=1000000007;
const int maxn=1000005;

int g[1005][1005];
int turn[1005][1005];
int n,m;
int dir[4][2]={1,0,-1,0,0,1,0,-1};

struct node{
    int x,y;
    int turn;
    int dir;
};

node st,en;

void bfs(){
    queue<node> q;
    node u;
    u.x=st.x;u.y=st.y;u.turn=0;u.dir=-5;
    turn[u.x][u.y]=0;
    q.push(u);
    
    while(!q.empty()){
        u=q.front();q.pop();
        
        node v;
        for(int i=0;i<4;i++){
            v.x=u.x+dir[i][0];
            v.y=u.y+dir[i][1];
            v.turn=u.turn;
            v.dir=u.dir;
            
            if(u.x==en.x&&u.y==en.y&&u.turn<=2){
                printf("YES
");
                return;
            }
            if(v.dir!=i&&v.dir!=-5) v.turn++;
            
            if(v.x==en.x&&v.y==en.y&&v.turn<=2){
                printf("YES
");
                return;
            }
            
            if(v.x<1||v.x>n||v.y<1||v.y>m||g[v.x][v.y]!=0) continue;
                    
            if(v.turn>2) continue;
        
            if(v.turn<=turn[v.x][v.y]){
                turn[v.x][v.y]=v.turn;
                v.dir=i;
                q.push(v);
            }        
        }    
    }
    printf("NO
");
}

int main(){
    while(scanf("%d %d",&n,&m)!=EOF&&n&&m){
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++) cin>>g[i][j];
        }
        
        int q;
        scanf("%d",&q);
        while(q--){
            for(int i=1;i<=n;i++)
             for(int j=1;j<=m;j++) turn[i][j]=1000000;
            scanf("%d %d %d %d",&st.x,&st.y,&en.x,&en.y);
            
            if(g[st.x][st.y]!=g[en.x][en.y]||!g[st.x][st.y]||!g[en.x][en.y]) printf("NO
");
            else bfs();
        }    
    }
    return 0;
}