计算疫情扩散的时间

• 计算疫情扩散时间 | 时间限制：1秒 | 内存限制：32768K | 语言限制：不限
  在一个地图中(地图由n*n个区域组成），有部分区域被感染病菌。感染区域每天都会把周围（上下左右）的4个区域感染。

请根据给定的地图计算，多少天以后，全部区域都会被感染。

如果初始地图上所有区域全部都被感染，或者没有被感染区域，返回-1
收起
输入描述:
一行N*N个数字（只包含0,1，不会有其他数字）表示一个地图，数字间用,分割，0表示未感染区域，1表示已经感染区域
每N个数字表示地图中一行，输入数据共表示N行N列的区域地图。
例如输入1,0,1,0,0,0,1,0,1，表示地图
1,0,1
0,0,0
1,0,1
输出描述:
一个整数，表示经过多少天以后，全部区域都被感染
示例1
输入
1,0,1,0,0,0,1,0,1
输出
2
说明
1天以后，地图中仅剩余中心点未被感染；2天以后，全部被感染。
示例2
输入
0,0,0,0
输出
-1
说明
无感染区域
示例3
输入
1,1,1,1,1,1,1,1,1
输出
-1
说明
全部都感染
备注:
1<=N<200

解法：改点是否会被感染，取决于改点的周围的四个点，是否有1 存在。
代码如下：

def spread_time(string):
    string_list=string.split(",")
    length = len(string_list)
    if string_list.count("1") == length or string_list.count("0") ==length:
        return -1
    slice = int(pow(length,0.5))
    rank = [string_list[i*slice:(i+1)*slice] for i in range(slice)]
    rank1 =[string_list[i*slice:(i+1)*slice] for i in range(slice)]
    def spread(rank,rank1):
        for i in range(len(rank)):
            for j in range(len(rank)):
                if rank[i][j] == "1":
                    continue
                if i > 0:
                    if rank1[i - 1][j] == "1":
                        rank[i][j] = "1"
                if j > 0:
                    if rank1[i][j - 1] == "1":
                        rank[i][j] = "1"
                if 0 < j < len(rank) - 1:
                    if rank1[i][j + 1] == "1":
                        rank[i][j] = "1"
                if 0 < i < len(rank) - 1:
                    if rank1[i + 1][j] == "1":
                        rank[i][j] = "1"
    days = 0
    while True:
        spread(rank,rank1)
        days+=1
        if [j for i in rank for j in i].count("1") == length:
            return days
        temp = rank
        rank = rank1
        rank1 = temp

if __name__ == '__main__':
    print(spread_time("1,0,1,0,0,0,1,0,1"))
    print(spread_time("1,1,1,1,1,1,1,1,1"))