HDU - 1175 连连看

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

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

 

基本就是道简单bfs吧，可能难点就是确定转弯。为了这个先写一个ff表示现在的方向，再写一个num统计转弯的次数就行。

注意，0代表的是该位置没有棋子，所以如果查询的位置是0直接否。

再吐槽一句，正常连连看都是绕开棋子走的好吗，这个完全反过来了。

AC代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <map>
#include <stack>
#include <string>
#include <queue>
using namespace std;
int s[1010][1010];
struct node
{
    int x,y,num,ff;
    bool operator <(const node &a)const
    {
        return num>a.num;
    }
};

int dir[4][2]={1,0,0,1,0,-1,-1,0};
int vis[1010][1010];
int main()
{
    int n,m;
    while(cin>>n>>m&&n+m)
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
        {
            scanf("%d",&s[i][j]);
        }

    int t;
    cin>>t;
    while(t--)
    {
        node a,b;
        scanf("%d%d%d%d",&a.x,&a.y,&b.x,&b.y);
        if((s[a.x][a.y]!=s[b.x][b.y])||s[a.x][a.y]==0||s[b.x][b.y]==0)
        {
            cout<<"NO\n";
            continue;
        }

        memset(vis,0,sizeof(vis));
        int flag=0;
        a.num=-1;
        a.ff=-1;
        vis[a.x][a.y]=1;
        priority_queue<node>q;
        while(!q.empty())
            q.pop();
        q.push(a);
        while(!q.empty())
        {
            node c;
            c=q.top();
            q.pop();
            for(int i=0;i<4;i++)
            {
                node d;
                d.x=c.x+dir[i][0];
                d.y=c.y+dir[i][1];
                if(c.ff!=i)
                {
                    d.ff=i;
                    d.num=c.num+1;
                }
                else
                {
                    d.ff=c.ff;
                    d.num=c.num;
                }
                if(d.num>2)
                    continue;
                if((s[d.x][d.y]==0||(s[d.x][d.y]==s[b.x][b.y]&&d.x==b.x&&d.y==b.y))&&d.x>=1&&d.x<=n&&d.y>=1&&d.y<=m&&!vis[d.x][d.y])
                {
                    vis[d.x][d.y]=1;
                    if(d.x==b.x&&d.y==b.y&&d.num<=2)
                    {
                        flag=1;
                        break;
                    }
                    q.push(d);
                }
            }
            if(flag)
                break;
        }
        flag?printf("YES\n"):printf("NO\n");
    }
}
return 0;
}