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+3;
int dir[4][2]= {{0,1},{0,-1},{-1,0},{1,0}};//上下左右
int vis[maxn][maxn],a[maxn][maxn],in[maxn][maxn],sum[maxn*maxn+10];
int n,m,ans=1;
struct node
{
    int x,y;
};
inline void bfs(int x,int y,int k)
{
    ans=1;
    node s;
    s.y=y,s.x=x;
    vis[x][y]=1;
    queue <node> Q;
    Q.push(s);
    while (!Q.empty())
    {
        node now=Q.front();
        Q.pop();
        in[now.x][now.y]=k;
        for (int i=0; i<4; i++)
        {
            node ne;
            ne.x=now.x+dir[i][0];
            ne.y=now.y+dir[i][1];
            if (ne.x>=1&&ne.x<=n&&ne.y>=1&&ne.y<=n&&!vis[ne.x][ne.y]&&a[ne.x][ne.y]!=a[now.x][now.y])
            {
                vis[ne.x][ne.y]=1;
                ans++;
                Q.push(ne);
            }
        }
    }
    sum[k]=ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
    {
        for (int j=1;j<=n;j++)
        {
            scanf("%1d",&a[i][j]);
        }
    }
    int k=0;
    for (int i=1;i<=n;i++)
    {
        for (int j=1;j<=n;j++)
        {
            if(in[i][j]==0)
            {
                bfs(i,j,++k);
            }
        }
    }
    for (int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d\n",sum[in[x][y]]);
    }
    return 0;
}

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

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