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。

#include<bits/stdc++.h>
using namespace std;
struct pos
{
    int x;
    int y;
};
int mark[1005][1005],pic[1005][1005],dx[]= {0,0,1,-1},dy[]= {1,-1,0,0},n,a,m;
queue<pos>q,t;
void bfs(int sx,int sy)
{
    int tx,ty;
    int x,y;
    a=1;
    q.push(pos{sx,sy});
    t.push(pos{sx,sy});
    mark[sx][sy]=-1;
    while(!q.empty())
    {
        x=q.front().x;
        y=q.front().y;
        q.pop();
        for(int i=0; i<4; i++)
        {
            tx=x+dx[i];
            ty=y+dy[i];
            if(tx<1||ty<1||tx>n||ty>n||mark[tx][ty]||pic[x][y]==pic[tx][ty])continue;
            q.push(pos{tx,ty});
            t.push(pos{tx,ty});
            mark[tx][ty]=-1;
            a++;
        }
    }
    while(!t.empty())
    {
        mark[t.front().x][t.front().y]=a;
        t.pop();
    }
}
int main()
{
    int a,b;
    memset(pic,0,sizeof(pic));
    memset(mark,0,sizeof(mark));
    cin>>n>>m;
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=n; j++)
        {
            scanf("%1d",&pic[i][j]);
        }
    }
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=n; j++)
        {
            if(!mark[i][j])bfs(i,j);
        }
    }
    while(m--)
    {
        scanf("%d%d",&a,&b);
        printf("%d\n",mark[a][b]);
    }
    return 0;
}