import scala.collection.mutable.HashMap

​

class Node() {

  var children: HashMap[Character, Node] = new HashMap[Character, Node]()

  var isWord:Boolean = false

​

}

​

class WordDictionary() {

  var root:Node = new Node

​

​

  def insertWord(word: String): Unit = {

    var node = this.root

    for (c <- word.toCharArray) {

      if (!node.children.contains(c))

        node.children(c) = new Node

      node = node.children(c)

    }

    node.isWord = true

  }

​

  def searchWord(word: String): Boolean = {

    var node = this.root

    for (c <- word.toCharArray) {

      if (!node.children.contains(c)) return false

      node = node.children(c)

    }

    node.isWord

  }

  def startsWith(word: String): Boolean = {

    var node = this.root

    for (c <- word.toCharArray) {

      if (!node.children.contains(c)) return false

      node = node.children(c)

    }

    true

  }

​

}

​

object Main {

  def main(args: Array[String]): Unit = {

    println("Hello World!")

    val keys = Array("the", "a", "there", "answer", "any", "by", "bye", "their", "abc")

    println("Keys to insert: ")

    for(k <- keys)

      print(k + ", ")

    println()

    val d = new WordDictionary

    for (i <- 0 until keys.length) {

      d.insertWord(keys(i))

    }

    println("Searching 'there' in the dictionary results: " + d.searchWord("there"))

  }

}