01迷宫

题目描述

有一个仅由数字000与111组成的n×nn \times nn×n格迷宫。若你位于一格0上，那么你可以移动到相邻444格中的某一格111上，同样若你位于一格1上，那么你可以移动到相邻444格中的某一格000上。

你的任务是：对于给定的迷宫，询问从某一格开始能移动到多少个格子（包含自身）。

输入输出格式

输入格式：

第111行为两个正整数n,mn,mn,m。

下面nnn行，每行nnn个字符，字符只可能是000或者111，字符之间没有空格。

接下来mmm行，每行222个用空格分隔的正整数i,ji,ji,j，对应了迷宫中第iii行第jjj列的一个格子，询问从这一格开始能移动到多少格。

输出格式：

mmm行，对于每个询问输出相应答案。

输入输出样例

输入样例#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。这题数据太强了，要做各种处理，我交了几十次才完美AC。。难受

 

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<cstdio>
#include<unordered_map>
using namespace std;
typedef pair<int,int> PII;
const int N=1010;
char g[N][N];
int n,m,ans[1000001],d;
int dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};
int stx[N][N];
int bfs(int a, int b){
    int res=0;
    queue<PII>q;
    q.push({a,b});
    while(q.size()){
        auto t=q.front();
        q.pop();
        if(stx[t.first][t.second])continue;
        res++;
        stx[t.first][t.second]=d;
        for(int i=0;i<4;i++){
            int x=t.first+dx[i],y=t.second+dy[i];
            if(x<0||x>=n||y<0||y>=n)continue;
            if((g[t.first][t.second]=='0'&&g[x][y]=='1')||(g[t.first][t.second]=='1'&&g[x][y]=='0')){
                    q.push({x,y});
                }
        }
    }
    ans[d]=res;
    return res;
}
int main(void){
    scanf("%d%d", &n, &m);
    for(int i=0;i<n;i++)scanf("%s",g[i]);
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            if(!stx[i][j]){
                d++;
                bfs(i,j);
            }
    for(int i=0,a,b;i<m;i++){
        scanf("%d%d", &a, &b);
        a--,b--;
        cout<<ans[stx[a][b]]<<endl;
    }
    return 0;
}