hdu1175——连连看

bfs思路。

加一个mark数组，用来判断是否入队。其初始化为3。

注意一点：如果想用取余来判断转向的话，很容易错，会把反方向当做不变。

#include<iostream> #include<cstring> #include<cstdio> using namespace std; int g[1005][1005]; int dir[][2]={{0,-1},{-1,0},{0,1},{1,0}}; int n,m; class node { public: int x,y; int step; int di; }; #define maxn 1000008 node queue[maxn]; int front,rear; int mark[1005][1005]; void solve(int x1,int y1,int x2,int y2) { int i,j; front=0;rear=0; if(g[x1][y1]!=g[x2][y2]||g[x1][y1]==0||g[x2][y2]==0||(x1==x2&&y1==y2)) printf("NO\n"); else { for(i=1;i<=n;i++) for(j=1;j<=m;j++) mark[i][j]=3; front=0;rear=0; for(i=0;i<4;i++) { rear++; queue[rear].x =x1;queue[rear].y=y1; queue[rear].di =i; queue[rear].step =0; } mark[x1][y1]=0; while(front !=rear) { front=(front+1)%maxn; for(i=0;i<4;i++) { int tempx=queue[front].x +dir[i][0]; int tempy=queue[front].y +dir[i][1]; if(tempx<=n&&tempx>0&&tempy<=m&&tempy>0&&(g[tempx][tempy]==0||(tempx==x2&&tempy==y2))) { int k=queue[front].step ; if(queue[front].di !=i) k++; if(k>mark[tempx][tempy]) continue; mark[tempx][tempy]=k; rear=(rear+1)%maxn; queue[rear].x =tempx;queue[rear].y=tempy; queue[rear].step =k;queue[rear].di =i; if(tempx==x2&&tempy==y2&&queue[rear].step <=2) { printf("YES\n"); return ; } } } } printf("NO\n"); } } int main() { int i,j; while(cin>>n>>m&&n&&m) { for(i=1;i<=n;i++) for(j=1;j<=m;j++) scanf("%d",&g[i][j]); int q,x1,y1,x2,y2; cin>>q; while(q--) { cin>>x1>>y1>>x2>>y2; solve(x1,y1,x2,y2); } } return 0; }