[HDOJ1175]连连看

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=1175

连连看

Time Limit: 20000/10000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 23190    Accepted Submission(s): 5718

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
 

 

之前用DFS写了一次，内存超了。于是用BFS重写了一次。

 

注意满足条件的判断，这个稍微有一点复杂。应该有初始情况两个数不相等的情况。

注意题目要求“线的转折次数不超过两次”，我没有算转折点，我算的折线数目。（≤3）

 #include <stdio.h>
 #include <string.h>
 #include <queue>
 #include <iostream>
 using namespace std;
 int n, m, s;
 int map[][];
 int vis[][];
 int x1, y1, x2, y2;
 int dir[][]={{,},{-,},{,},{,-}};
 typedef struct Position
 {
     int x, y;
     int count;
 }Position;

 int BFS()
 {
     queue<Position> q;
     Position fst, tmp;
     fst.x = x1;
     fst.y = y1;
     fst.count = -; //init. first one not calculate.
     q.push(fst);
     vis[x1][y1] = ;
     int sum;
     while(!q.empty())
     {
         fst = q.front();
         q.pop();
         for(int i = ; i < ; i++)
         {
             tmp.x = fst.x + dir[i][];
             tmp.y = fst.y + dir[i][];
             if(fst.count == i)
             {
                 sum = vis[fst.x][fst.y];
             }
             else
             {
                 sum = vis[fst.x][fst.y] + ;
             }
             tmp.count = i;
             if( tmp.x >=  &&
                 tmp.x < n  &&
                 tmp.y >=  &&
                 tmp.y < m  &&
               (vis[tmp.x][tmp.y] ==  || sum <= vis[tmp.x][tmp.y]))
             {
                vis[tmp.x][tmp.y] = sum;
                if(tmp.x == x2 && tmp.y == y2)
                 {
                     if(vis[tmp.x][tmp.y] <= )
                     {
                         return ;
                     }
                 }
                 if(vis[tmp.x][tmp.y] > )
                 {
                     continue;
                 }
                 if(map[tmp.x][tmp.y] == )
                 {
                     q.push(tmp);
                 }
             }
         }
     }
     return ;
 }

 int main()
 {
     while(scanf("%d%d",&n,&m) != EOF && n+m)
     {
         for(int i = ; i < n; i++)
         {
             for(int j = ; j < m; j++)
             {
                 scanf("%d", &map[i][j]);
             }
         }
         scanf("%d", &s);
         for(int i =  ; i < s; i++)
         {
             scanf("%d %d %d %d", &x1, &y1, &x2, &y2);
             x1--;
             x2--;
             y1--;
             y2--;
             if((map[x1][y1] != map[x2][y2]) ||
                 map[x1][y1] ==  ||
                 map[x2][y2] ==  ||
                 (x1==x2&&y1==y2))
             {
                 printf("NO\n");
                 continue;
             }
             memset(vis, , sizeof(vis));
             if(BFS())
             {
                 printf("YES\n");
             }
             else
             {
                 printf("NO\n");
             }
         }

     }
     return ;
 }