洛谷——P1141 01迷宫

Posted 2020-10-08

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。

 

一直wa  7个点找了半天也不知道哪错了，结果发现是读入的时候出了问题ORZ

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 5000
using namespace std;
char a[N][N];
bool vis[N][N],g[N][N];
int n,m,x,y,ans,f[N][N];
int xx[4]={0,0,1,-1},yy[4]={1,-1,0,0};
int read()
{
    int x=0,f=1; char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1; ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘; ch=getchar();}
    return x*f;
}
void dfs(int x,int y)
{
    if(x<1||y<1||x>n||y>n||vis[x][y]||g[x][y]) return ; 
    if(!g[x][y]) ans++;
    vis[x][y]=true,g[x][y]=true;
    for(int i=0;i<4;i++)
    {
        int fx=x+xx[i],fy=y+yy[i];
        if(a[fx][fy]!=a[x][y])
         dfs(fx,fy);
    }
    vis[x][y]=false;
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
          cin>>a[i][j];
     } 
    while(m--)
    {
        x=read(),y=read();
        ans=0;
        memset(g,0,sizeof(g));
        memset(vis,0,sizeof(vis));
        dfs(x,y);
        printf("%d\n",ans);
    }
    return 0;
}

爆搜70

正解：记忆化搜索

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 5000
using namespace std;
char a[N][N];
bool vis[N][N],g[N][N];
int n,m,x,y,ans,cnt=1,d[1000005],f[N][N];
int xx[4]={0,0,1,-1},yy[4]={1,-1,0,0};
int read()
{
    int x=0,f=1; char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1; ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘; ch=getchar();}
    return x*f;
}
void dfs(int x,int y)
{
    if(x<1||y<1||x>n||y>n||f[x][y]) return ; 
    f[x][y]=cnt;ans++;
    for(int i=0;i<4;i++)
    {
        int fx=x+xx[i],fy=y+yy[i];
        if(a[fx][fy]!=a[x][y])
         dfs(fx,fy);
    }
    d[cnt]=ans;
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
          cin>>a[i][j];
     } 
    while(m--)
    {
        x=read(),y=read();
        if(!f[x][y])
        {
            ans=0;
            dfs(x,y);cnt++;
        }
        ans=d[f[x][y]];
        printf("%d\n",ans);
    }
    return 0;
}

记忆化搜索