连连看

连连看

 

Problem Description

“连连看”相信很多人都玩过。没玩过也没关系，下面我给大家介绍一下游戏规则：在一个棋盘中，放了很多的棋子。如果某两个相同的棋子，可以通过一条线连起来（这条线不能经过其它棋子），而且线的转折次数不超过两次，那么这两个棋子就可以在棋盘上消去。不好意思，由于我以前没有玩过连连看，咨询了同学的意见，连线不能从外面绕过去的，但事实上这是错的。现在已经酿成大祸，就只能将错就错了，连线不能从外围绕过。
玩家鼠标先后点击两块棋子，试图将他们消去，然后游戏的后台判断这两个方格能不能消去。现在你的任务就是写这个后台程序。
 

Input

输入数据有多组。每组数据的第一行有两个正整数n,m(0 < n <= 1000,0 < m < 1000)，分别表示棋盘的行数与列数。在接下来的n行中，每行有m个非负整数描述棋盘的方格分布。0表示这个位置没有棋子，正整数表示棋子的类型。接下来的一行是一个正整数q(0 < q < 50)，表示下面有q次询问。在接下来的q行里，每行有四个正整数x1,y1,x2,y2,表示询问第x1行y1列的棋子与第x2行y2列的棋子能不能消去。n=0,m=0时，输入结束。
注意：询问之间无先后关系，都是针对当前状态的！
 

Output

每一组输入数据对应一行输出。如果能消去则输出”YES”,不能则输出”NO”。
 

Sample Input

3 4
1 2 3 4
0 0 0 0
4 3 2 1
4
1 1 3 4
1 1 2 4
1 1 3 3
2 1 2 4
3 4
0 1 4 3
0 2 4 1
0 0 0 0
2
1 1 2 4
1 3 2 3
0 0

Sample Output

YES
NO
NO
NO
NO
YES

 

代码

#include<iostream>
using namespace std;
const int maxn=1005,inf=1<<30;
int n,m,Map[maxn][maxn],wan[maxn][maxn];
bool flag;
int q,sx,sy,ex,ey;
int dir[][2]={
{0,1},{-1,0},{0,-1},{1,0}
};
void dfs(int x,int y,int cnt,int f)
{
    if(x==ex&&y==ey)
    {
        flag=1;
       // cout<<"cnt = "<<cnt<<endl;
        return ;
    }
    for(int i=0;i<4&&!flag;i++)
    {
        int nx=x+dir[i][0],ny=y+dir[i][1];
        if(1<=nx&&nx<=n&&1<=ny&&ny<=m&&(Map[nx][ny]==0||(nx==ex&&ny==ey)))
        {
            if(i==f||f==-1)
            {
                wan[nx][ny]=cnt;
                dfs(nx,ny,cnt,i);
                if(flag) return ;
            }
            else if(cnt<2&&wan[nx][ny]>wan[x][y])
            {
                wan[nx][ny]=cnt+1;
                dfs(nx,ny,cnt+1,i);
                if(flag) return ;
            }
        }
    }
}
int main()
{
    while(cin>>n>>m&&(n+m))
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
            cin>>Map[i][j];
        cin>>q;
        for(int i=0;i<q;i++)
        {
            cin>>sx>>sy>>ex>>ey;
            if(Map[sx][sy]!=Map[ex][ey]) {cout<<"NO"<<endl;continue;}
            if(Map[sx][sy]==0) {cout<<"NO"<<endl;continue;}
            flag=0;
            fill(&wan[0][0],&wan[maxn][0],inf);
            wan[sx][sy]=0;
            dfs(sx,sy,0,-1);
            if(flag) cout<<"YES"<<endl;
            else cout<<"NO"<<endl;
        }
    }
    return 0;
}