01迷宫（BFS+记忆)

题目描述

有一个仅由数字 0 与 1 组成的 n × n 格迷宫。若你位于一格 0 上，那么你可以移动到相邻 4 格中的某一格 1 上，同样若你位于一格 1 上，那么你可以移动到相邻 4 格中的某一格 0 上。
你的任务是：对于给定的迷宫，询问从某一格开始能移动到多少个格子（包含自身）。

输入

输入的第 1 行为两个正整数 n，m （n ≤ 1000，m ≤ 100000）
下面 n 行，每行 n 个字符，字符只可能是 0 或者 1，字符之间没有空格。
接下来 m 行，每行 2 个用空格分隔的正整数 i,j，对应了迷宫中第 i 行第 j 列的一个格子，询问从这一格开始能移动到多少格。

输出

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

样例输入

2 2
01
10
1 1
2 2

样例输出

4
4

---

分析

• 一般看到这道题都会第一时间想到直接暴力搜索。是的没错，无论是DFS还是BFS都能帮我们找到结果，但是现在的问题就在于m的取值。若m取到极限值100000，我们真的要去搜十万次，肯定会TLE，怎么解决这个问题呢？
  
  
• 我们仔细想想，如果迷宫中A、B两格是相通的，那么我们是不是就可以从A走到B，也可以从B走到A。由此我们可以想到，如果迷宫中的某一块是连通的，那么在这一块中的任意一格，它们能移动到的格子数是相同的！！
  
  
• 因此我们可以在一次搜索中，把这次能走到的格子都做上标记，当下一次要搜索这些格子时，直接用上一次的结果。（记忆化）
  
  
  
  
  
  
  
  

代码如下：

#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
struct point{
    int x,y;
    point(int a,int b):x(a),y(b){}
};
const int maxn=1000+5;
char smap[maxn][maxn];//存储地图
int d[8]={0,1,0,-1,1,0,-1,0};
int mem[maxn][maxn];//给搜索过的位置作标记
int n,m,i,j;
int ans[maxn*100]={0};//存储第t次搜索的结果
int bfs(int x,int y,int t)//t表示第几次搜索
{
    int cou=1;
    queue<point> q;
    point p(x,y);
    q.push(p);
    while(!q.empty())
    {
        p=q.front();
        //cout<<p.x<<" "<<p.y<<endl;
        q.pop();
        for(i=0;i<8;i+=2)
        {
            int tx=p.x+d[i];
            int ty=p.y+d[i+1];
            if(tx>0&&tx<=n&&ty>0&&ty<=n&&smap[p.x][p.y]!=smap[tx][ty]&&mem[tx][ty]==0)
            {
                point p1(tx,ty);
                q.push(p1);
                mem[tx][ty]=t;
                cou++;
            }
        }
    }
    return cou;
}
int main()
{
    cin>>n>>m;
    for(i=1;i<=n;i++)
        cin>>smap[i]+1;
    for(int k=1;k<=m;k++)
    {
        int x,y;
        cin>>x>>y;
        if(!mem[x][y])//如果该格子没有被搜索过，就要搜索一遍
        {
            mem[x][y]=k;
            ans[k]=bfs(x,y,k);
            cout<<ans[k]<<endl;
        }
        else//被搜索过的话，直接输出结果
        {
            int t=mem[x][y];
            ans[m]=ans[t];
            cout<<ans[m]<<endl;
        }
    }
    return 0;
}

最后再附上简单的bfs的模板

template <class T>
void bfs(T a)//有没有返回值看情况
{
    queue<T> q;
    q.push(a);
    while(!q.empty())
    {
        T p=q.front();//用一个临时变量存储队列的第一份元素
        q.pop();
        for(i=0;i<8;i+=2)//在目前这点的四周移动
        {
            T temp = p+dx[i];//移动操作
            if(temp in the map&& no visit)//移动后的点，在地图内，而且还没有被访问过
            {
                point p1(tx,ty);
                q.push(p1);
                memory[][]=true;//访问过的点进行标记
            }
        }//通过一次循环操作，能把周围的能到达的点都压进队列
    }
}