2023华为od机试真题B卷Python【矩阵里的非一元素】

题目

假设存在一个m*n的二维数组，其成员取值范围为0、1和2。1具有同化特性，它具体的同花步骤是：每经过1秒，如果其上下左右值为0，那么修改为1。而值为2的元素不会有任何变化。初始时，将数组的[0,0]元素修改为1。经过足够长的时间后，需要计算矩阵中有多少个元素的取值为0或2（即0和2数量之和）。

输入描述：

输入的前两个数字是矩阵的大小。后面是数字矩阵的内容。

输出描述：

返回矩阵中非1的元素个数。

示例1：

输入：

4 4
0 0 0 0
0 2 2 2
0 2 0 0
0 2 0 0

输出：

9

说明：

输入的前两个数字表示矩阵的大小，后面的数字表示矩阵的内容。初始时，将起始位置(0,0)修改为1。经过一段时间后，矩阵变为：

1 1 1 1
1 2 2 2
1 2 0 0
1 2 0 0

因此，矩阵中非1的元素个数为9。

代码

from collections