问题 J: 新冠病毒的传播

题目描述

最近新冠病毒疫情非常严重，由于我们国家采取了有力的措施，才没有使疫情进一步的扩大。今天，作为计算机专业的学生，我们来用程序模拟一下各种情况下的新冠病毒传播情况。现在给定一个n * m的网格，每个网格可以有以下三个值之一：
值 0 代表隔离带。
值 1 代表健康人群。
值 2 代表感染人群。
每天，任何与感染人群（在 4 个正方向上）相邻的健康人都会感染。如果遇到隔离带，病毒就会被阻断。
输出直到单元格中没有健康人为止所必须经过的最小天数。如果不可能所有人都被感染，输出 -1。

输入

测试数据由多组测试样例组成。第一行输入两个正整数 n ( 1 <= n <= 500 ） 和 m ( 1 <= m <= 500 ），
接下来输入n * m个数字，数字均由 0 ， 1 ，2构成

输出

每组测试样例，输出直到单元格中没有健康人为止所必须经过的最小天数。如果不可能，输出 -1。

样例输入 Copy

3 3
2 1 1
1 1 0
0 1 1
3 3
2 1 1
0 1 1
1 0 1
1 2
0 2

样例输出 Copy

4
-1
0

思路

这是一道BFS题，借鉴了他人的代码，琢磨了好几遍才写出来，第一个是队列不会用，其次是不知道如何增加天数。首先创建一个结构体，包含二维数组的下标以及天数，将输入进来的感染人群压入队列，进行bfs。定义一个临时的结构体变量赋值为队首，对它的四个方向进行搜索，如果发现健康人群则将他压入队列，他的天数也就是临时变量的后一天，然后将他改变为感人人群。

AC代码

#include <cstdio>
#include <iostream>
#include <queue>
using namespace std;
int n,m,ans=0;
int map[505][505];
int dir[4][2]={{1,0},{0,1},{-1,0},{0,-1}};
struct fun;
queue<fun> q;
struct fun{
	int x;
	int y;
	int day;
}virus;
void bfs(){
	while(q.size()){
		fun temp=q.front();
		if(temp.day>ans){
			ans=temp.day;
		}
		q.pop();
		for(int i=0;i<4;i++){
			int tx=temp.x+dir[i][0];
			int ty=temp.y+dir[i][1];
			if(tx<0||ty<0||tx>=n||ty>=m){
				continue;
			}
			if(map[tx][ty]==1){
				fun k;
				k.x=tx;
				k.y=ty;
				k.day=temp.day+1;
				q.push(k);
				map[tx][ty]=2;
			}
		}
	}
	return ;
}

int main(){
	while(scanf("%d %d",&n,&m)!=EOF){
		ans=0;
		for(int i=0;i<n;i++){
			for(int j=0;j<m;j++){
				scanf("%d",&map[i][j]);
				if(map[i][j]==2){
					virus.x=i;
					virus.y=j;
					virus.day=0;
					q.push(virus);
				}
			}
		}
		bfs();
		for(int i=0;i<n;i++){
			for(int j=0;j<m;j++){
				if(map[i][j]==1){
					ans=-1;
					break;
				}
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}