计算疫情扩散时间 JavaScript

区域被感染病菌

在一个地图中(地图由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

function infect(nums){
  let step = Math.sqrt(nums.length),len = nums.length;
  if(nums.reduce((pre,cur)=>pre+cur,0)==len||nums.reduce((pre,cur)=>pre+cur,0)==0){
    return -1
  }
  let day = 0
  //扩散
  while(nums.reduce((pre,cur)=>pre+cur,0)!=len){
    let arrfn = []
    for(let i =0;i<len;i++){
      if(nums[i]==1){
        arrfn.push(i)
      }
    }
    arrfn.forEach((item)=>{
      spread(item)
    })
    day ++
  }
  function spread(index){
    //找出第几行，第几列
    //上
    if(0<=index-step<len&&nums[index-step]==0){
      nums[index-step]=1
    }
    //下
    if(0<=index+step<len&&nums[index+step]==0){
      nums[index+step]=1
    }
    //左
    if(index%step>=1&&nums[index-1]==0){
      nums[index-1]=1
    }
    //右
    if((index+1)%step!=0&&nums[index+1]==0){
      nums[index+1]=1
    }
  }
  return day;
}