P1141 01迷宫 题解

01迷宫

题目描述

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

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

输入格式

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

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

输出格式

$m$ 行，对于每个询问输出相应答案。

样例 #1

样例输入 #1

2 2
01
10
1 1
2 2

样例输出 #1

4
4

提示

所有格子互相可达。

对于 $20%$ 的数据，$n \leq 10$；

对于 $40%$ 的数据，$n \leq 50$；

对于 $50%$ 的数据，$m \leq 5$；

对于 $60%$ 的数据，$n,m \leq 100$；

对于 $100%$ 的数据，$n \leq 1000,m \leq 100000$。

---

思路分析

对于每次询问均进行bfs，会产生路径的重复搜索，会超时，因此采用记忆化搜索。

对于每次询问，如果该点先前未被搜索过，则从该点开始进行bfs，并在bfs后将本次搜索经过的所有点对应的答案全部更新(在搜索时随便用一个STL储存经过的点，搜完之后遍历)；如果先前被搜索过，则直接输出答案即可。

---

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
char map[1010][1010];
int n,m,ans[1010][1010],dx[4]={1,-1,0,0},dy[4]={0,0,1,-1};
bool a[1010][1010];
struct Point{
    int x,y;
};
void read(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++){
        scanf(" %c",&map[i][j]);
    }
}
void bfs(int x0,int y0){
    int cnt=0;
    queue<Point> q,q0;
    q.push({x0,y0});
    while(!q.empty()){
        q0.push(q.front());
        cnt++;
        int x_now=q.front().x,y_now=q.front().y;
        a[x_now][y_now]=true;
        q.pop();
        for(int i=0;i<4;i++){
            int x=x_now+dx[i],y=y_now+dy[i];
            if(x>=1&&x<=n&&y>=1&&y<=n&&map[x][y]!=map[x_now][y_now]&&!a[x][y]){
                q.push({x,y});
                a[x][y]=true;
            }
        }
    }
    while(!q0.empty()){
        int x=q0.front().x,y=q0.front().y;
        ans[x][y]=cnt;
        q0.pop();
    }
}
void solve(){
    read();
    for(int i=0;i<m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        if(ans[x][y]==0){
            bfs(x,y);
        }
        cout<<ans[x][y]<<endl;
    }
}
int main(){
    solve();
    return 0;
}