“流感传染”问题

流感传染

描述

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

输入

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

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

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

输出

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

此代码不是为了解答此题。因为vs2015不支持变长数组。不能满足题目要求。
只是为了显示一种思路。

#include<iostream>
#include <iomanip>
using namespace std;
 
const int N = 5;                //定义宿舍边长
char dorm[N][N];                 //定义一个N*N的全局 二维字符数组
 
int function(int a)
{
    int count = 0;                                             //用来记录所有宿舍共有多少个感染者
    for (int d = 1; d < a; d++)
    {
        for (int i = 0; i < N; i++)
        {
            for (int j = 0; j < N; j++)
            {
                if (dorm[i][j] == '@')                         //找到数组中感染的人“@”。把其上下左右都标为“*”，以避免多算上新患者，避免使新患者周围也被传染。
                {
                    if (i + 1 < N && dorm[i + 1][j] != '#' && dorm[i + 1][j] != '@')   //考虑边界和 空房子的情况
                        dorm[i + 1][j] = '*';
                    if (j + 1 < N && dorm[i][j + 1] != '#' && dorm[i][j + 1] != '@')
                        dorm[i][j + 1] = '*';
                    if (i - 1 >= 0 && dorm[i - 1][j] != '#' && dorm[i - 1][j] != '@')
                        dorm[i - 1][j] = '*';
                    if (j - 1 >= 0 && dorm[i][j - 1] != '#' && dorm[i][j - 1] != '@')
                        dorm[i][j - 1] = '*';
                }
            }
        }
        for (int k = 0; k < N; k++)                    //一次传染完成后，将新传染的“*”赋值为“@”
        {
            for (int h = 0; h < N; h++)
            {
                if (dorm[k][h] == '*')
                    dorm[k][h] = '@';
            }
        }
 
    }
    for (int i = 0; i < N; i++)                               //统计感染者人数
        for (int j = 0; j < N; j++)
        {
            if (dorm[i][j] == '@')
                count++;
        }
    return count;
}
 
int main()
{
    cout << "请输入宿舍感染情况矩阵:" << endl;
    for (int i = 0; i < N; i++)
        for (int j = 0; j < N; j++)
            cin >> dorm[i][j];
    int day;                                      //输入第几天
    cout << "请输入第几天:" << endl;
    cin >> day;
    cout << "此时感染人数:" << function(day) << endl;
    cout << "此时感染情况" << endl;
    for (int i = 0; i < N; i++)                   //输出此时的宿舍感染分布情况
    {
        for (int j = 0; j < N; j++)
            cout << dorm[i][j] << ' ';
        cout << endl;
    }
 
    return 0;
}