P1141 01迷宫

https://www.luogu.org/problemnew/show/P1141

题目描述

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

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

输入输出格式

输入格式：

输入的第1行为两个正整数n，m。

下面n行，每行n个字符，字符只可能是0或者1，字符之间没有空格。

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

输出格式：

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

输入输出样例

输入样例#1： 

2 2
01
10
1 1
2 2

输出样例#1： 

4
4

说明

所有格子互相可达。

对于20%的数据，n≤10；

对于40%的数据，n≤50；

对于50%的数据，m≤5；

对于60%的数据，n≤100，m≤100；

对于100%的数据，n≤1000，m≤100000。

其实思路不是很复杂 一堆题解都用并查集写的 但是用联通块就可以写 （求联通块dfs写起来比bfs快多了。。）

相同联通块的点的数量是一样多的 所以只要枚举的时候查一下有没有被book就行了 所有的计数操作在输入询问的时候进行 可以省很多事情

关键是用临时数组把搜过的点记录下来 每次联通块搜完更新一次 然后注意m的范围是1e6

#include<bits/stdc++.h>

using namespace std;
#define MAXN 1000

char mp[1005][1005];
int Next[4][2]= {0,1,1,0,0,-1,-1,0},book[1000005][2],vis[1005][1005],n,m,a,b,sum;

void dfs(int x,int y)
{
    sum++;
    book[sum][0]=x;//标记坐标的数组很关键 相当于维护每个点所在联通块的点的数量
    book[sum][1]=y;
    vis[x][y]=1;
    for(int i=0; i<4; i++)
    {
        int nx=x+Next[i][0];
        int ny=y+Next[i][1];
        if(mp[nx][ny]==mp[x][y] || vis[nx][ny]>0 || nx<1 || ny<1 || nx>n || ny>n)
            continue;
        dfs(nx,ny);
    }
}

int main()
{
    cin>>n>>m;
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
            cin>>mp[i][j];
    for(int i=1; i<=m; i++)
    {
        cin>>a>>b;
        sum=0;
        if(vis[a][b]>0)
        {
            cout<<vis[a][b]<<endl;
            continue;
        }
        dfs(a,b);
        for(int i=1; i<=sum; i++)
            vis[book[i][0]][book[i][1]]=sum; //这里更新联通块点的数量
        cout<<sum<<endl;
    }
    return 0;
}