华为OD机试 Python 水域污染

题目

我们需要计算水域污染的速度。一开始有一个一个N*N的数组，表示水域。请计算多少天后水域将被全部污染。

0表示纯净的水域，1表示已经污染的水域。每个格子每天只能污染上下左右相邻的水域。

例如：

示例输入：

1,0,1,0,0,0,1,0,1

将其转化为数组：

1 0 1
0 0 0
1 0 1

输出：

2

样例解释：

第一天后，水域变为：

1 1 1
1 0 1
1 1 1

第二天，水域被全部污染。

代码

from collections import deque

def calculate_days_to_pollute_all(input_str):
    elements =