2020小米网络选拔赛I Walking Machine
这道题一看就是bfs,然后超时了。
仔细想想判断每个点是否能走出边界,调用了mn次bfs函数。
接着我这个菜鸡就开始反向思维,从边缘可以走出地图的点反向bfs,求出可以到达的点的数量。
最多只需要调用2(m+n-1)次bfs。
#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
const int N=100005;
int vis[1005][1005],vis1[1005][1005],dx[4]={-1,1,0,0},dy[4]={0,0,-1,1},ans;
char g[1005][1005];
struct team {
int x,y;
team (int a,int b):x(a),y(b) {};
};
queue<team> q;
void bfs(int x,int y) {
queue<team> t1;
t1.push(team(x,y));
while(!t1.empty()) {
team t=t1.front();t1.pop();
for(int i=0;i<4;i++) {
int xx=t.x+dx[i],yy=t.y+dy[i];
if(vis[xx][yy]) continue;
if(i==0&&g[xx][yy]=='S') {
vis[xx][yy]=1;
ans++;
t1.push(team(xx,yy));
}
if(i==1&&g[xx][yy]=='W') {
vis[xx][yy]=1;
ans++;
t1.push(team(xx,yy));
}
if(i==2&&g[xx][yy]=='D') {
vis[xx][yy]=1;
ans++;
t1.push(team(xx,yy));
}
if(i==3&&g[xx][yy]=='A') {
vis[xx][yy]=1;
ans++;
t1.push(team(xx,yy));
}
}
}
}
int main() {
ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
int n,m;
cin>>n>>m;
memset(vis,0,sizeof(vis));
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
cin>>g[i][j];
ans=0;
for(int i=1;i<=m;i++) {
if(g[1][i]=='W') ans++,vis[1][i]=1,q.push(team(1,i));
if(g[n][i]=='S') ans++,vis[n][i]=1,q.push(team(n,i));
}
for(int i=1;i<=n;i++) {
if(g[i][1]=='A') ans++,vis[i][1]=1,q.push(team(i,1));
if(g[i][m]=='D') ans++,vis[i][m]=1,q.push(team(i,m));
}
while(!q.empty()) {
team t=q.front();q.pop();
bfs(t.x,t.y);
}
cout<<ans<<'\n';
return 0;
}
/*
*/