P1141 01迷宫（连通块模板）

题目描述

有一个仅由数字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

说明/提示

所有格子互相可达。

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

解题思路：看完题目明显是搜索，但是一开始写了一个bfs直接T了，想了一下发现，其实在同一个连通块，所以直接给连通块标号，每个点记录一下在几号连通块内，记得数组大小开maxn*maxn，考虑每个人一个连通块的情况。

bfs版本，慢一点

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e3+;
int dir[][]= {{,},{,-},{-,},{,}};//上下左右
int vis[maxn][maxn],a[maxn][maxn],in[maxn][maxn],sum[maxn*maxn+];
int n,m,ans=;
struct node
{
    int x,y;
};
inline void bfs(int x,int y,int k)
{
    ans=;
    node s;
    s.y=y,s.x=x;
    vis[x][y]=;
    queue <node> Q;
    Q.push(s);
    while (!Q.empty())
    {
        node now=Q.front();
        Q.pop();
        in[now.x][now.y]=k;
        for (int i=; i<; i++)
        {
            node ne;
            ne.x=now.x+dir[i][];
            ne.y=now.y+dir[i][];
            if (ne.x>=&&ne.x<=n&&ne.y>=&&ne.y<=n&&!vis[ne.x][ne.y]&&a[ne.x][ne.y]!=a[now.x][now.y])
            {
                vis[ne.x][ne.y]=;
                ans++;
                Q.push(ne);
            }
        }
    }
    sum[k]=ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=;i<=n;i++)
    {
        for (int j=;j<=n;j++)
        {
            scanf("%1d",&a[i][j]);
        }
    }
    int k=;
    for (int i=;i<=n;i++)
    {
        for (int j=;j<=n;j++)
        {
            if(in[i][j]==)
            {
                bfs(i,j,++k);
            }
        }
    }
    for (int i=;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d\n",sum[in[x][y]]);
    }
    return ;
}

dfs版本，简单易写，速度快

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e3+;
int dir[][]= {{,},{,-},{-,},{,}};//上下左右
int vis[maxn][maxn],a[maxn][maxn],in[maxn][maxn],sum[maxn*maxn+];
int n,m,ans=;
struct node
{
    int x,y;
};
inline void dfs(int x,int y,int k)
{
    in[x][y]=k;
    vis[x][y]=;
    ans++;
    for (int i=;i<;i++)
    {
        int nx=x+dir[i][],ny=y+dir[i][];
        if (nx>=&&nx<=n&&ny>=&&ny<=n&&!vis[nx][ny]&&a[nx][ny]!=a[x][y])
            dfs(nx,ny,k);
    }
    sum[k]=ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=;i<=n;i++)
    {
        for (int j=;j<=n;j++)
        {
            scanf("%1d",&a[i][j]);
        }
    }
    int k=;
    for (int i=;i<=n;i++)
    {
        for (int j=;j<=n;j++)
        {
            if(in[i][j]==)
            {
                dfs(i,j,++k);
                ans=;
            }
        }
    }
    for (int i=;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d\n",sum[in[x][y]]);
    }
    return ;
}