洛谷 P1141 记忆化搜索

https://www.luogu.org/problemnew/show/P1141

题目描述

有一个仅由数字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%20\%20%的数据，n≤10n≤10n≤10；

对于40%40\%40%的数据，n≤50n≤50n≤50；

对于50%50\%50%的数据，m≤5m≤5m≤5；

对于60%60\%60%的数据，n≤100,m≤100n≤100,m≤100n≤100,m≤100；

对于100%100\%100%的数据，n≤1000,m≤100000n≤1000,m≤100000n≤1000,m≤100000。

思路：看数据范围，很明显的记忆化搜索。一次dfs所能达到的所有地方构成一个联通块，且此连通块内的任意一个方块的答案都相同。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<algorithm>
using namespace std;
typedef pair<int,int> P;

int d[4][2]={{-1,0},{1,0},{0,1},{0,-1}};
int dp[1005][1005];
int s[1005][1005];
int vis[1005][1005];
char b[1005][1005];
vector<P> vec;
int n,m;
int cnt;

void dfs(int x,int y)
{
    P p(x,y);
    vec.push_back(p);
    vis[x][y]=1;
    for(int i=0;i<4;i++)
    {
        int dx=x+d[i][0];
        int dy=y+d[i][1];
        if(dx>0&&dx<=n&&dy>0&&dy<=n&&!vis[dx][dy]&&(s[x][y]^s[dx][dy])==1)
        {
            ++cnt;
            dfs(dx,dy);
        }
    }
}
int main()
{
    memset(dp,-1,sizeof(dp));
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%s",b[i]+1);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            s[i][j]=b[i][j]-48;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(!vis[i][j])
            {
                cnt=1;
                dfs(i,j);
                int num=vec.size();
                while(num)
                {
                    --num;
                    dp[vec[num].first][vec[num].second]=cnt;
                    vec.pop_back();
                }
            }
        }
    }
    int x,y;
    for(int i=0;i<m;i++)
    {
        scanf("%d %d",&x,&y);
        printf("%d\n",dp[x][y]);
    }
    return 0;
}