HDU1157 连连看

HDU1157 连连看DFS
测试样例：
样例1：
3 2
0 1
0 2
0 1
1
1 2 3 2
YES

样例2：
1 3
2 2 2
1
1 1 3 3
NO

主要是解决初始方向问题就OK了

#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
struct V{
    int x,y;
};
int a[1001][1001],visited[1001][1001];
int n,m;
void Read(){
    scanf("%d%d",&n,&m);
    if(n==0&&m==0)exit(0);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            scanf("%d",&a[i][j]);
}
int Ju(int x,int y){//不越界
    if(x>=1&&x<=n&&y>=1&&y<=m)return 1;
    else return 0;
}
int x1,y1,x2,y2;
int flag;
int add_x[4]={1,-1,0,0};
int add_y[4]={0,0,-1,1};//四个方向
void dfs(V v,int change,int last,int deep){//change为方向改变次数，last记录上一个方向，deep为递归深度
    if(flag||change>2)return;
    if(v.x==x2&&v.y==y2){//不可用a[x][y]==a[x2][y2]替代
        flag=1;
        return;
    }
    for(int i=0;i<4;i++){
        if(deep==0)last=i;//关键1，当递归深度为0也就是刚进来时没有确定方向，4个方向任选
        int x=v.x+add_x[i],y=v.y+add_y[i];
        if(Ju(x,y)&&visited[x][y]==0&&
           (a[x][y]==0||x==x2&&y==y2)){//关键2，x==x2&&y==y2不可用a[x][y]==a[x2][y2]替代，
                                        //因为可能有样例2
            visited[x][y]=1;
            if(last==i)dfs({x,y},change,last,deep+1);//判断方向是否改变
            else dfs({x,y},change+1,i,deep+1);
            if(flag)return;
            visited[x][y]=0;
        }
    }
}
int main(){
    int q;
    while(1){
        Read();
        scanf("%d",&q);
        while(q--){
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            if(!(x1==x2&&y1==y2)&&//起终点位置不可相同
               Ju(x1,y1)&&Ju(x2,y2)&&
               a[x1][y1]==a[x2][y2]&&a[x1][y1]){//起终点数值相等且不可是0
                memset(visited,0,sizeof(visited));
                flag=0;
                dfs({x1,y1},0,0,0);//关键3，这里传入了的方向0，其实初始方向不可确定，传入0是错误的，
                                   //但是总得有个传入值吧，这个错误不必担心，它被关键1修补了
                                   //如样例1
                if(flag){printf("YES\n");continue;}
            }
            printf("NO\n");
        }
    }
}