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≤100000n1000,m100000。

解题思路:看完题目明显是搜索,但是一开始写了一个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
",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
",sum[in[x][y]]);
    }
    return 0;
}
原文地址:https://www.cnblogs.com/ztdf123/p/11367616.html