洛谷P1141 01迷宫

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

 

 

【思路】

  BFS。

  只要相邻节点颜色不同而且未访问则拓展。随读入随操作，如果节点未标记则以该节点拓展。c表示节点颜色，cnt表示颜色的数目。

  注意数组大小一定要开够了，cnt的大小为maxn * maxn。

 

【代码】

 

 

#include<iostream>
#include<queue>
using namespace std;

const int maxn = 1000+10;
const int dx[]={0,1,0,-1};
const int dy[]={1,0,-1,0};
struct Node{
    int x,y;
};
int G[maxn][maxn];
int c[maxn][maxn];
int cnt[maxn*maxn];

int n,m;
int colors=1; 

inline bool inside(int x,int y) {
    return x>=0 && x<n && y>=0 && y<n;
}
void bfs(int s,int t) {
    queue<Node> q;
    q.push((Node){s,t});
    c[s][t]=colors; cnt[colors]++;
    while(!q.empty()) 
    {
        Node u=q.front(); q.pop();
        int x=u.x , y=u.y;
        for(int i=0;i<4;i++)
        {
            int xx=x+dx[i],yy=y+dy[i];
            if(inside(xx,yy) && (G[x][y]!=G[xx][yy]) && c[xx][yy]==0) 
            {
                   c[xx][yy]=colors; cnt[colors]++;
                   q.push((Node){xx,yy});
            }
        }
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin>>n>>m;
    char ch;
    for(int i=0;i<n;i++)
      for(int j=0;j<n;j++) {
             cin>>ch;
             G[i][j]=ch-'0';
      }
    int x,y;
    for(int i=1;i<=m;i++) {
        cin>>x>>y;
        x--; y--;
        colors=i;
        if(c[x][y]==0) bfs(x,y);
        cout<<cnt[c[x][y]]<<"\n";
    }
    return 0;
}