题目描述
有一个仅由数字 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
2 2
01
10
1 1
2 2
样例输出 #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;
}
---------------------------------------------------------------------------------------------------------------------------------
最开始我写的忘记memset了。当时超时了,所以定义了l数组,l[i][j]代表i行j列的位置能到多少个地方,因为有很多连通块,可以在dfs一遍后遍历,如果vis是true就把他的l也改为cnt。
不过这道题的数据卡的很死,所以还是不行,需要把bool二维数组vis改为一个set,这样遍历时可以用auto优化了。