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

样例输入 #1

样例输出 #1

4
4

提示

所有格子互相可达。

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

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

---------------------------------------------------------------------------------------------------------------------------------

#include <iostream>
#include <cstring>
#include <set>
using namespace std;

int dis[4][2]={{0,-1},{0,1},{1,0},{-1,0}};
set<int> hash1;
char a[1010][1010];
int l[1010][1010];
int n,m,sx,sy,cnt;
void dfs(int x,int y)
{
   int k = (y<<16)|x;
   hash1.insert(k);
   for(int i=0;i<4;i++)
   {
       int nx=x+dis[i][0];
       int ny=y+dis[i][1];
       if(l[nx][ny]||a[nx][ny]==a[x][y])
           continue;
       int nk=(ny<<16)|nx;
       if( hash1.count(nk))
           continue;
       dfs(nx,ny);
   }
}
int main()
{
   cin>>n>>m;
   for(int i=1;i<=n;i++)
       for(int j=1;j<=n;j++)
       {
           char c;
           cin>>c;
           a[i][j]=c;
       }
   for(int i=0;i<=n+1;++i){
       l[i][0] = l[i][n+1] = 1;
   }
   for(int i=0;i<=n+1;++i){
       l[0][i] = l[n+1][i] = 1;
   }
   while(m--)
   {
       cin>>sx>>sy;
       if(l[sx][sy]!=0)
       {
           cout<<l[sx][sy]<<endl;
           continue;
       }
       hash1.clear();
       dfs(sx,sy);
       int cnt=hash1.size();

       for(auto k : hash1) {
           int x = k &0xffff;
           int y = (k>>16)&0xffff;
           l[x][y] = cnt;
       }

       l[sx][sy]=cnt;
       cout<<cnt<<endl;
   }
   return 0;
}