代码如下:
#include<bits/stdc++.h>
using namespace std;
const int N=110;
char g[N][N],bg[N][N];
int dx[]={-1,1,0,0},dy[]={0,0,-1,1};
int main()
{
int n;
cin>>n;
for(int i=0;i<n;i++) cin>>g[i];
int m;
cin>>m;
m--;
while(m--)
{
memcpy(bg,g,sizeof g);
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
if(g[i][j]!='@') continue; //没有感染
for(int k=0;k<4;k++) //枚举周围4个点
{
int a=i+dx[k],b=j+dy[k];
if(a<0||a>=n||b<0||b>=n) //越界
continue;
if(bg[a][b]=='#') //空白
continue;
bg[a][b]='@'; //感染
}
}
}
memcpy(g,bg,sizeof bg); //将第i天当成第i+1天的前一天
}
int cnt=0;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
{
if(g[i][j]=='@')
cnt++;
}
cout<<cnt;
return 0;
}