A simple trie

mytrie.rb

module Spellchecker
  class Trie
    attr_reader   :kids
    attr_accessor :terminal

    def initialize()
      @kids = Hash.new
      @terminal = false
    end

    # Add word to the trie
    def add(word)
      if word != "" 
        cur = self
        word.downcase.each_char do |character|
          cur.kids[character] = Trie.new if not cur.kids.has_key? character
          cur = cur.kids[character]
        end
        cur.terminal = true
      end
    end

    # Check if word is in trie
    def has_word?(word)
      cur = self
      word.each_char do |character|
        return false if not cur.kids.has_key? character
        cur = cur.kids[character]
      end
      cur.terminal
    end
  end
end