解题思路:多起点的bfs问题。由于有步数限制,因此需要在入队前后做下步数检查。
#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
struct node
{
int x,y;
};
int n,m,X,v[1005][1005];
int dx[8]= {0,0,1,-1,1,1,-1,-1};
int dy[8]= {1,-1,0,0,1,-1,1,-1};
char a[1005][1005];
int bfs()
{
queue<node>q;
int i,j,k,cnt=0;/**< cnt计数器,一发现感染就自增 */
for(i=1; i<=n; i++)
for(j=1; j<=m; j++)
if(a[i][j]=='@')
q.push({i,j}),v[i][j]=1;
while(q.size())
{
int x=q.front().x,y=q.front().y;
q.pop();
cnt++;/**< 出队时计数器自增 */
if(v[x][y]==X+1) /**< 第X天被感染的不再入队,用于步数控制 */
continue;
for(k=0; k<8; k++)
{
int nx=x+dx[k],ny=y+dy[k];//迷宫三要素
if(nx>=1&&nx<=n&&ny>=1&&ny<=m&&a[nx][ny]=='*'&&v[nx][ny]==0)
{
v[nx][ny]=v[x][y]+1;
q.push({nx,ny});
}
}
}
return cnt;
}
int main()
{
ios::sync_with_stdio(0),cin.tie(0);
int i,j;
cin>>n>>m>>X;
for(i=1; i<=n; i++)
for(j=1; j<=m; j++)
cin>>a[i][j];
cout<<bfs();
return 0;
}