#include<bits/stdc++.h>
using namespace std;
const int MAXN=1010;
int a[MAXN][MAXN]={};
//int cnt=0,maxcnt=0;
int ty[4]={1,0,-1,0};
int tx[4]={0,1,0,-1};
int vis[MAXN][MAXN];
int n,m;
int st,ed;
void dfs(int x,int y,int &cnt){
cnt++;
for(int i=0;i<4;i++){
if(a[x][y]!=a[x+tx[i]][y+ty[i]]&&vis[x+tx[i]][y+ty[i]]==0&&(x+tx[i]>=0&&x+tx[i]<n&&y+ty[i]>=0&&y+ty[i]<n)){
vis[x+tx[i]][y+ty[i]]=1;
dfs(x+tx[i],y+ty[i],cnt);
}
}
}
int main()
{
freopen("in.txt","r",stdin);freopen("out.txt","w",stdout);
cin>>n>>m;
for(int i=0;i<n;i++){
string s;cin>>s;
for(int j=0;j<n;j++){
a[i][j]=s[j]-'0';
}
}
for(int i=0;i<m;i++){
fill(vis[0],vis[0]+MAXN*MAXN,0);
int sx,sy;cin>>sx>>sy;
st=sx,ed=sy;int cnt=0;
vis[sx-1][sy-1]=1;
dfs(sx-1,sy-1,cnt);
cout<<cnt<<endl;cnt=0;//maxcnt=0;
}
return 0;
}
上面一段时70分代码,2,9,10三个点超时,下面的一段代码使用了_out数组记录连通块中的每一个元素,因为连通块中的每一个元素为起点最终的cnt都是一样的,这样省去了很多时间,注意_out数组要开的大一点,因为cnt可能会很大。
#include<bits/stdc++.h>
using namespace std;
const int MAXN=1010;
int a[MAXN][MAXN]={};
int ty[4]={1,0,-1,0};
int tx[4]={0,1,0,-1};
int vis[MAXN][MAXN];
int n,m;
int ans[1000000][2];
int _out[MAXN][MAXN];
void dfs(int x,int y,int &cnt){
cnt++;
ans[cnt][0]=x;ans[cnt][1]=y;
for(int i=0;i<4;i++){
if(a[x][y]!=a[x+tx[i]][y+ty[i]]&&vis[x+tx[i]][y+ty[i]]==0&&(x+tx[i]>=0&&x+tx[i]<n&&y+ty[i]>=0&&y+ty[i]<n)){
vis[x+tx[i]][y+ty[i]]=1;
dfs(x+tx[i],y+ty[i],cnt);
}
}
}
int main()
{
freopen("in.txt","r",stdin);//freopen("out.txt","w",stdout);
cin>>n>>m;
for(int i=0;i<n;i++){
string s;cin>>s;
for(int j=0;j<n;j++){
a[i][j]=s[j]-'0';
}
}
fill(vis[0],vis[0]+MAXN*MAXN,0);
fill(ans[0],ans[0]+MAXN*MAXN,0);
for(int i=0;i<n;i++){
for(int t=0;t<n;t++){
if(vis[i][t]==0){
int cnt=0;
vis[i][t]=1;
dfs(i,t,cnt);
for(int j=1;j<=cnt;j++){
_out[ans[j][0]][ans[j][1]]=cnt;
}
}
}
}
for(int i=0;i<m;i++){
int sx,sy;cin>>sx>>sy;
cout<<_out[sx-1][sy-1]<<endl;
}
return 0;
}