Update 1345-jump-game-iv.js

Author: everthis

1345-jump-game-iv.js

@@ -68,16 +68,20 @@ const minJumps = function (arr) {
     const len = queue.length
     for(let i = 0; i < len; i++) {
       const cur = queue.shift()
-      visited.add(cur)
       if(cur === n - 1) return distance
-      if(cur + 1 < n && !visited.has(cur + 1)) queue.push(cur + 1)
-      if(cur - 1 >= 0 && !visited.has(cur - 1)) queue.push(cur - 1)
-      for(let next of indexMap.get(arr[cur])) {
-        if(!visited.has(next)) queue.push(next)
+      const tmp = indexMap.get(arr[cur])
+      tmp.push(cur - 1)
+      tmp.push(cur + 1)
+      for(let e of tmp) {
+        if(e >= 0 && e < n && !visited.has(e)) {
+          visited.add(e)
+          queue.push(e)
+        }
       }
       indexMap.set(arr[cur], [])
     }
     distance++
   }
   return -1
 }
+