PTA 感染人数 python

设某住宿区域是一个n×n的方阵，方阵中的每个小方格为一个房间，房间里可能住一个人，也可能空着。第一天，某些房间中住着的人得了一种高传染性的流感，以后每一天，得流感的人会使其邻居（住在其上、下、左、右方向存在的房间里面的人）传染上流感，请问：第m天总共有多少人得流感？

输入格式:

第一行输入两个整数n,m(1<n≤20,1≤m≤100)，含义如上述；接着输入n行，每行n个字符，表示住宿区域第一天的房间情况，其中，@表示当天该房间住着得流感的人，.表示该房间住着健康的人，#表示该房间是空的。

输出格式:

输出一个整数，表示第m天得了流感的人数。

输入样例1:

5 3
#....
.....
...##
.#.@.
@.#..

输出样例1:

10

输入样例2:

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

输出样例2:

16

def spread_infection(matrix, n, m):
    # Convert the matrix of strings to a matrix of characters
    matrix = [list(row) for row in