empty * case enhancement

homuhe · Dec 4, 2016 · 0ee4455 · 0ee4455
1 parent b0d068d
commit 0ee4455
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/scala/com/ir/Trie.scala b/src/main/scala/com/ir/Trie.scala
@@ -196,7 +196,8 @@ object Trie {
       if (input.forall(char => (char - 97) >= 0 && (char - 97)  < 25 || char == '*')) {
 
         if (input.count(_ == '*') == 1) {
-          if (trie.contains(input.filter(_ != '*'))) //empty '*' case
+          if (trie.contains(input.filter(_ != '*')) && input.startsWith("*") //empty '*' case
+                                                    || input.endsWith("*"))
             println(input.filter(_ != '*'))
           query(input).foreach(println)
         }