HD连连看

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

思考：这道题主要就用到bfs套用；
需要考虑输出结果的几个可能性，
1.两个数字不同输出NO，
2.两个数字是同一个输出NO
3.其中一个数字为0输出NO
4.转折不能超过两次
*********************************``

我的代码

#include <iostream>
#include <cstdio>
#include <queue>
using namespace std;
#define INF 10000000
int xy[4][2] = { { -1 , 0 } , { 1 , 0 } , { 0 , -1 } , { 0 , 1 } };
int m, n;
int x1, x2, y1, y2;
int a[1005][1005], b[1005][1005];
class data
{
public:
    int x, y;
    int k, dir;
};
int bfs()
{
    if((x1 == x2 && y1 == y2) || (a[x1][y1] != a[x2][y2]) || !a[x1][y1] || !a[x2][y2])
        return 0;
    queue <data> que;
    data te, temp;
    te.x = x1; te.y = y1;
    te.k = 0; te.dir = -1;
    que.push(te);
    b[te.x][te.y] = 0;
    while(!que.empty())
    {
        temp = que.front();
        que.pop();
        if(temp.x == x2 && temp.y == y2)
            return 1;
        for(int k = 0; k < 4; k++)
        {
            te.x = temp.x + xy[k][0];
            te.y = temp.y + xy[k][1];
            if(temp.dir < 0)
            {
                te.dir = k;
                te.k = 0;
            }
            else if(temp.dir == k)
            {
                te.dir = temp.dir;
                te.k = temp.k;
            }
            else
            {
                te.dir = k;
                te.k = temp.k + 1;
            }
            if(te.x >= 1 && te.x <= m && te.y >= 1 && te.y <= n && te.k <= 2 && te.k <= b[te.x][te.y]) 
            {
                if(a[te.x][te.y] == 0 || (te.x == x2 && te.y == y2))
                {
                    b[te.x][te.y] = te.k;
                    que.push(te);
                }
            }
        }
    }
    return 0;
}
 
int main()
{
    int i, j, num;
    while(scanf("%d%d", &m, &n) != EOF)
    {
        if(m == 0 && n == 0)
            break;
        for(i = 1; i <= m; i++)
        {
            for(j = 1; j <= n; j++)
                scanf("%d", &a[i][j]);
        }
        scanf("%d", &num);
        while(num--)
        {
            scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
            for(i = 0; i <= m; i++)
            {
                for(j = 0; j <= n; j++)
                    b[i][j] = INF;
            }
            if(bfs())
                printf("YES\n");
            else
                printf("NO\n");
        }
    }
    return 0;
}