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

这道题宽搜只能70分。。。
所以用宽搜+记忆化(不是纯记忆化,只是思路一样)
代码还是很好理解的,注意不能用memset,否则会TLE

#include<bits/stdc++.h>
int n,m,a[1010025][3],cnt,aa[100005],k;
int x[4]={0,1,0,-1},y[4]={1,0,-1,0},d[1005][1005];
bool b[1005][1005];
inline void bfs(int i,int j){
    cnt++;d[i][j]=k;
    int head=1,foot=2;
    a[2][1]=i,a[2][2]=j;
    while(head<foot){
        head++;
        for(register int q=0;q<=3;q++){
            int xx=a[head][1]+x[q];
            int yy=a[head][2]+y[q];
            if(xx>0 && xx<=n && yy>0 && yy<=n && !d[xx][yy]
            && b[a[head][1]][a[head][2]]!=b[xx][yy]){
                    cnt++;
                    foot++;
                    a[foot][1]=xx;
                    a[foot][2]=yy;
                    d[xx][yy]=k;
               }
        }
    }
    aa[k]=cnt;
}
int main(){
    scanf("%d%d",&n,&m);
    for(register int i=1;i<=n;i++)
        for(register int j=1;j<=n;j++){
            char c;
            scanf(" %c",&c);
            if(c=='1')  b[i][j]=1;
        }
    for(register int i=1;i<=m;i++){
        int xxx,yyy;
        scanf("%d%d",&xxx,&yyy);
        k++;
        if(d[xxx][yyy])   cnt=aa[d[xxx][yyy]];
        else   bfs(xxx,yyy);
        printf("%d\n",cnt);
        cnt=0;
    }
}
posted @ 2017-11-29 18:46  Monster_Qi  阅读(96)  评论(0编辑  收藏  举报