#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; }
文章来源: P1141 01迷宫