6262:流感传染——暴力搜索

描述

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

输入

第一行一个数字n，n不超过100，表示有n*n的宿舍房间。
接下来的n行，每行n个字符，’.’表示第一天该房间住着健康的人，’#’表示该房间空着，’@’表示第一天该房间住着得流感的人。
接下来的一行是一个整数m，m不超过100.

输出

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

样例输入

5
....#
.#.@.
.#@..
#....
.....
4

样例输出

16

 

思路

直接递推每天会被传染的人，注意要区分易感染者和新感染者，可以用不同符号来存储。

代码

#include <iostream>
#include <cstring>
using namespace std;

char ch[105][105];
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
int n,m;

bool in(int x,int y){
	if(x<1||y<1||x>n||y>n) return false;
	return true;
}

int main(){
	cin>>n;
	int count=0;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>ch[i][j];
		}
	}
	cin>>m;
	for(int i=1;i<m;i++){
		for(int j=1;j<=n;j++){
			for(int k=1;k<=n;k++){
				if(ch[j][k]=='@'){
					for(int l=0;l<4;l++){
						int xx=j+dx[l];
						int yy=k+dy[l];
						if(in(xx,yy)){
							if(ch[xx][yy]=='.'){
								ch[xx][yy]='!';
							}
						}
					}
				}
				
			}
		}
		for(int k=1;k<=n;k++){
			for(int j=1;j<=n;j++){
				if(ch[k][j]=='!') ch[k][j]='@';
			}
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(ch[i][j]=='@') count++;
		}
	}
	cout <<count<<endl;
	return 0;
}