1191：流感传染

1191：流感传染

【题目描述】
有一批易感人群住在网格状的宿舍区内，宿舍区为n*n的矩阵，每个格点为一个房间，房间里可能住人，也可能空着。在第一天，有些房间里的人得了流感，以后每天，得流感的人会使其邻居传染上流感，（已经得病的不变），空房间不会传染。请输出第m天得流感的人数。

【输入】
第一行一个数字n，n不超过100，表示有n*n的宿舍房间。

接下来的n行，每行n个字符，’.’表示第一天该房间住着健康的人，’#’表示该房间空着，’@’表示第一天该房间住着得流感的人。

接下来的一行是一个整数m，m不超过100。

【输出】
输出第m天，得流感的人数。

【输入样例】
5
…#
.#.@.
.#@…
#…
…
4
【输出样例】
16

代码

#include <iostream>
using namespace std;

const int LN=100;
int a[LN][LN];

int fn_1(int n,int m){
	if(m<=0) return 0;
	int ans=0;
	for(int i=0;i<n;i++ )
		for(int j=0;j<n;j++)
		{
			if( a[i][j]==1 &&(
				(i-1>=0 && a[i-1][j]==0) ||
				(i+1<n && a[i+1][j]==0) ||
				(j-1>=0 && a[i][j-1]==0) ||
				(j+1<n && a[i][j+1]==0))){
				a[i][j]=-1;
				ans++;	
			}
		}
		for(int i=0;i<n;i++ )
		for(int j=0;j<n;j++)
		{
			if(a[i][j]==-1) a[i][j]=0;
		}	
	return ans;	
}

int main(){
	int n,m,f1=0;
	cin>>n;
	
	char ch;
	for(int i=0;i<n;i++)
		for(int j=0;j<n;j++){
			cin>>ch;
			if(ch=='.') a[i][j]=1;
			else if(ch=='#') a[i][j]=2;
			else if(ch=='@') {
				a[i][j]=0;
				f1++;
			}
		}	
	cin>>m;
	int fn;
	for(int i=2;i<=m;i++){
		fn=f1+fn_1(n,i-1);
		f1=fn;	
	}	
	cout<<f1<<endl;	
	return 0;			
}