洛谷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

说明

所有格子互相可达。

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

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

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

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

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

----------------------------------------------------------------------------------

连通分量

用个数组保存id为i的连通分量大小f[i]

 

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int N=1005;
inline int read(){
    char c=getchar(); int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return x;
}
int g[N][N],n,m,x,y;
int dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};
int f[N*N],vis[N][N],cnt=0,idcnt=0;
void dfs(int x,int y,int id){
    vis[x][y]=id;//printf("dfs%d %d %d\n",id,x,y);
    cnt++;
    for(int i=0;i<4;i++){
        int nx=x+dx[i],ny=y+dy[i];//printf("n i%d %d %d\n",i,nx,ny);
        if(nx>=1&&nx<=n&&ny>=1&&ny<=n&&vis[nx][ny]!=vis[x][y]&&g[nx][ny]!=g[x][y])
            dfs(nx,ny,id);
    }
}

int solve(int x,int y){
    if(vis[x][y]!=0) return f[vis[x][y]];
    cnt=0;
    dfs(x,y,++idcnt);
    return f[idcnt]=cnt;
}
int main(){
    n=read();m=read();
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++){
            char c=getchar();
            while(c!='0'&&c!='1') c=getchar();
            g[i][j]=c-'0';//cout<<g[i][j]<<" ";
        }
        
    for(int i=1;i<=m;i++){
        int x=read();y=read();
        printf("%d\n",solve(x,y));
        //cnt=0; memset(vis,0,sizeof(vis));
        //dfs(x,y,i);
        //printf("%d\n",cnt);
    }
    
}