洛谷-P1141 01迷宫

洛谷-P1141 01迷宫

原题链接:https://www.luogu.com.cn/problem/P1141


题目描述

有一个仅由数字0与1组成的\(n \times 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。

C++代码

#include <iostream>
#include <queue>
using namespace std;
#define MAXN 100000

struct node {
    int x;
    int y;
}p[MAXN];
int n,m,flag[1001][1001],ans[MAXN+1];
int dir[][4]={{-1,0},{0,1},{1,0},{0,-1}};
char a[1001][1001];
queue<node> q;

int bfs(node u, int k) {
    int i,j,sum=0;
    node v;
    q.push(u);
    while(!q.empty()) {
        u=q.front(),q.pop();
        if(flag[u.x][u.y])
            continue;
        for(i=0;i<4;++i) {
            v.x=u.x+dir[i][0];
            v.y=u.y+dir[i][1];
            if(v.x<1||v.x>n||v.y<1||v.y>n)
                continue;
            if(a[u.x][u.y]!=a[v.x][v.y])
                q.push(v);
        }
        flag[u.x][u.y]=k;
        ++sum;
    }
    return sum;
}

int main() {
    int i,j,k=1;
    cin>>n>>m;
    for(i=1;i<=n;++i)
        for(j=1;j<=n;++j)
            cin>>a[i][j];
    for(i=0;i<m;++i)
        cin>>p[i].x>>p[i].y;
    for(i=0;i<m;++i) {
        if(!flag[p[i].x][p[i].y]) {
            ans[k]=bfs(p[i],k);
            ++k;
        }
        cout<<ans[flag[p[i].x][p[i].y]]<<endl;
    }
    return 0;
}
上一篇:1001 害死人不偿命的(3n+1)猜想


下一篇:pandas_数据拆分与合并