HDU 1175 连连看(bfs + 剪枝)

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>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#define LL long long
using namespace std;
const int MAXN = 1000 + 10;
int n, m;
int M[MAXN][MAXN], vis[MAXN][MAXN];
struct Point
{
    int x, y;
    int dir, change;
};
Point S, E;
int dir[][2] = {{0,-1},{0,1},{-1,0},{1,0}};//上下左右四个方向
bool bfs()
{
    queue<Point>Q;
    S.dir = -1, S.change = 0;//起点方向设为-1
    Q.push(S);
    while(!Q.empty())
    {
        Point pre = Q.front();
        Q.pop();
        if(pre.x == E.x && pre.y == E.y)//已经和终点相连
            return true;
        for(int i=0;i<4;i++)
        {
            Point cnt;
            cnt.x = pre.x + dir[i][0];
            cnt.y = pre.y + dir[i][1];
            cnt.dir = i;
            cnt.change = pre.change;
            if(cnt.dir != pre.dir && pre.dir != -1) cnt.change++;
            if(cnt.x < 0 || cnt.x >= n || cnt.y < 0 || cnt.y >= m || cnt.change > 2)//判断是否越界和转弯次数
                continue;
            if(M[cnt.x][cnt.y] != 0 && !(cnt.x == E.x && cnt.y == E.y))//当前位置不为空
                continue;
            if(cnt.change < vis[cnt.x][cnt.y])
            {
                vis[cnt.x][cnt.y] = cnt.change;
                Q.push(cnt);
            }
        }
    }
    return false;
}
int main()
{
    while(scanf("%d%d", &n, &m)!=EOF)
    {
        if(n == 0 && m == 0)
            break;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
                scanf("%d", &M[i][j]);
        }
        int Q;
        scanf("%d", &Q);
        //Point S, E;
        while(Q--)
        {
            scanf("%d%d%d%d", &S.x, &S.y, &E.x, &E.y);
            S.x--;S.y--;E.x--;E.y--;
            if(S.x == E.x && S.y == E.y)//若起点和终点相同则输出NO
            {
                printf("NO\n");
                continue;
            }
            if(M[S.x][S.y] != M[E.x][E.y] || M[S.x][S.y] == 0 || M[E.x][E.y] == 0)
            {//起点和终点不同或者起点和终点有一个为空则输出NO
                printf("NO\n");
                continue;
            }
            for(int i=0;i<n;i++)
                for(int j=0;j<m;j++)
                    vis[i][j] = INT_MAX;
            if(bfs()) printf("YES\n");
            else printf("NO\n");
        }
    }
    return 0;
}