Update 1268-search-suggestions-system.js

Author: everthis

1268-search-suggestions-system.js

@@ -46,40 +46,46 @@ const suggestedProducts = function(products, searchWord) {
  * @param {string} searchWord
  * @return {string[][]}
  */
- const suggestedProducts = function(products, searchWord) {
+ const suggestedProducts = function (products, searchWord) {
+  products.sort()
   const root = new Node()
-  for(const str of products) {
+  for (const str of products) {
     addProduct(str)
   }
 
   const res = []
 
   let cur = root
-  for(const ch of searchWord) {
+  for (const ch of searchWord) {
     const tmp = []
-    if(cur == null) {
-        res.push(tmp)
-        continue
+    if (cur == null) {
+      res.push(tmp)
+      continue
     }
     const map = cur.children.get(ch)
-    if(map != null) {
-      const arr = [...map.words]
-      arr.sort()
-      tmp.push(...arr.slice(0, 3))
+    if (map != null) {
+      addThree(map.words.values(), tmp)
     }
 
     res.push(tmp)
     cur = map
   }
 
-
   return res
-  
+
+  function addThree(it, arr) {
+
+    for(let i = 0; i < 3; i++) {
+      const res = it.next()
+      if(res.value) arr.push(res.value)
+    }
+  }
+
   function addProduct(str) {
     let cur = root
-    for(const ch of str) {
+    for (const ch of str) {
       let next = cur.children.get(ch)
-      if(next == null) {
+      if (next == null) {
         next = new Node()
         cur.children.set(ch, next)
       }
@@ -88,12 +94,13 @@ const suggestedProducts = function(products, searchWord) {
     }
     cur.isWord = true
   }
-};
+}
 
 class Node {
   constructor() {
-      this.children = new Map()
-      this.words = new Set()
-      this.isWord = false
+    this.children = new Map()
+    this.words = new Set()
+    this.isWord = false
   }
 }
+