P1141 迷宫(记忆化BFS)

题目描述

有一个仅由数字00与11组成的n \times nn×n格迷宫。若你位于一格0上,那么你可以移动到相邻44格中的某一格11上,同样若你位于一格1上,那么你可以移动到相邻44格中的某一格00上。

你的任务是:对于给定的迷宫,询问从某一格开始能移动到多少个格子(包含自身)。

输入输出格式

输入格式:

 

第11行为两个正整数n,mn,m。

下面nn行,每行nn个字符,字符只可能是00或者11,字符之间没有空格。

接下来mm行,每行22个用空格分隔的正整数i,ji,j,对应了迷宫中第ii行第jj列的一个格子,询问从这一格开始能移动到多少格。

 

输出格式:

 

mm行,对于每个询问输出相应答案。

此题比较坑,交了三发才过,写博客纪念一下。400ms稳过。

#include<bits/stdc++.h>
using namespace std;

struct Node{
    int x,y,mov;
    Node(int _x,int _y,int _mov):x(_x),y(_y),mov(_mov){}
};
int next[4][2]={{1,0},{-1,0},{0,-1},{0,1}};
char c;
int n,m,arr[1010][1010],vis[1010][1010],x,y,ar[1001000][2],sum;//ar数组一定要开大一些,不然有一个点会wa

int bfs(int x,int y){
    sum=0;
    int s=0;
    queue<Node> q;
    while(!q.empty()){
        q.pop();
    }
    ar[sum][0]=x;
    ar[sum++][1]=y;
    q.push(Node(x,y,0));
    vis[x][y]=1;
    Node no=q.front();
    while(!q.empty()){
        no = q.front();
        q.pop();
        s++;
        for(int i=0;i<4;i++){
            x=no.x+next[i][0];
            y=no.y+next[i][1];
            if(!vis[x][y]&&(arr[x][y]!=arr[no.x][no.y])&&x>0&&x<=n&&y>0&&y<=n){
                ar[sum][0]=x;
                ar[sum++][1]=y;
                vis[x][y]=1;
                q.push(Node(x,y,no.mov));
            }
        }
    }
    return s;
}

int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cin>>c;
            if(c=='1') arr[i][j]=1;
            else arr[i][j]=0;
        }
    }
    for(int i=0;i<m;i++){
        cin>>x>>y;
        int a=0;
        if(!vis[x][y]){
            a = bfs(x,y);
            for(int i=0;i<sum;i++) {
                arr[ar[i][0]][ar[i][1]]=a;
                ar[i][0]=0;
                ar[i][1]=0;
            }
        }
        cout<<arr[x][y]<<endl;
    }
    return 0;
}