jtrie 0.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 2fe0bc8f28fbff700f9ba8ec762d43eaaaab312bea737cff08de26a928ed420a
+  data.tar.gz: a4c14c137644e08e12bd59b6f33471070ca65eaf25d88d6c3c0d10c0f08c2475
+SHA512:
+  metadata.gz: 6fad1ffac14eac381cf11c29e2d74ca3dfe9ef2258600d82c6409d8c7612326141465edd48ae2eb339351fde79312f3cd7a62df5c007f9844b5ebcef3db1082a
+  data.tar.gz: fbd20214bc17f3d57e2482e3f63ac861770e3a908d2394ae7747f57cab7cabaa605d3520a4e55f89b6a88ad87aa9058773215a569b7c38c17c6e797e4f714c3d

data/lib/jtrie.rb

@@ -0,0 +1,112 @@
+#require_relative 'csv_write'
+
+class TrieNode
+  attr_accessor :is_end, :children
+
+  def initialize(is_end = false)
+    @is_end = is_end
+    @children = Array.new(26)
+  end
+end
+
+class Jtrie
+  attr_accessor :root, :dictionary
+
+  def initialize
+    @root = TrieNode.new
+    @dictionay = []
+    @csv_dictionary = Csv.new
+    @test = []
+  end
+
+  def add(word)
+    @dictionay << word
+    # @test << word
+    word_chars = word.chars
+    current = @root
+    0.upto(word.length - 1) do |i|
+      letter = word_chars[i].ord - 'a'.ord
+      current.children[letter] = TrieNode.new if current.children[letter].nil?
+      current = current.children[letter]
+    end
+    current.is_end = true
+  end
+
+  def find(word)
+    word_chars = word.chars
+    current = @root
+    (0..word.length - 1).each do |i|
+      letter = word_chars[i].ord - 'a'.ord
+      return false if current.children[letter].nil?
+
+      current = current.children[letter]
+    end
+    current.is_end
+  end
+
+  def list(prefix)
+    result = []
+    @dictionay.each { |word| result << word if word.start_with?(prefix) }
+    result.length.zero? ? 'word is not in dictionary' : result
+  end
+
+  def include?(word)
+    find(word) { |found, root| return found && root.word }
+  end
+
+  def delete_word(word)
+    @dictionay.delete(word)
+    delete_recursively(word, 0, root)
+  end
+
+  # def write_csv
+  #   @csv_dictionary.save_to_csv(@test)
+  # end
+
+  # def read_csv
+  #   @csv_dictionary.read_from_csv
+  # end
+
+  private
+
+  def is_in_trie?(word, index, current)
+    if index == word.length
+      return false unless current.is_end
+
+      current.is_end = false
+      current.children.length.zero?
+    end
+  end
+
+  def delete_each(current, char, should_delete_ref)
+    if should_delete_ref
+      current.children.delete(char)
+      current.children.length.zero?
+    end
+  end
+
+  def delete_recursively(word, index, root)
+    current = root
+
+    is_in_trie?(word, index, current)
+
+    char = word[index]
+    return false unless current.children.include?(char)
+
+    next_node = current.children[char]
+    should_delete_ref = delete_recursively(word, index + 1, next_node)
+
+    delete_each(current, char, should_delete_ref)
+    false
+  end
+end
+
+# trie = Trie.new
+# trie.add('cat')
+# trie.add('camp')
+# trie.add('cave')
+# trie.add('cask')
+# trie.add('cow')
+# trie.add('apple')
+# trie.add('tester')
+# puts trie.list('')

metadata

@@ -0,0 +1,43 @@
+--- !ruby/object:Gem::Specification
+name: jtrie
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+platform: ruby
+authors:
+- Jeko Apulava
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2022-05-11 00:00:00.000000000 Z
+dependencies: []
+description: fast prefix tree implementation
+email: japulava@unisens.ge
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/jtrie.rb
+homepage: https://rubygems.org/gems/jtrie
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.32
+signing_key: 
+specification_version: 4
+summary: Prefix tree
+test_files: []