pretrie 0.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 7d72e5a7117dc00b96ddfb3aed12a90db76adfeeed454a4141c3fc752985dfb3
+  data.tar.gz: 91f22036f643489c27f691801da3e347e062ec05251e86af1f962b63d37a8f6c
+SHA512:
+  metadata.gz: 0d807b739d9fc0db111ba5b21c75a06f97e3b0663e4fd48ca3084f0916e080d22dad1bac105c6c6d78182246ca046e92a69cff4a189641caf5f86e57a47b18ac
+  data.tar.gz: 2d280c1ad2f81f5f6a00bf9f5ee24eb78a4145878e8121294e4b14d19e81d21fa5b07d6a68e2c8e50e3bf427e51d628a015b520da53678ff44ed45cf492781ab

data/lib/prefixtree/prefixtreenode.rb

@@ -0,0 +1,9 @@
+$LOAD_PATH << "../lib"
+class Node
+  attr_accessor :data, :is_word
+
+  def initialize
+    @data = {}
+    @is_word = false
+  end
+end

data/lib/prefixtree/tocsv.rb

@@ -0,0 +1,21 @@
+$LOAD_PATH << "../lib"
+require 'csv'
+
+class FileWriter
+  attr_accessor :path
+
+  def initialize
+    @path = 'lib/data.csv'
+  end
+
+  def save_to_csv(word_list)
+    CSV.open(path, 'a') do |csv|
+      csv << word_list
+    end
+    puts 'Successfully added to CSV'
+  end
+
+  def read_from_csv
+    CSV.read(path).to_a
+  end
+end

data/lib/prefixtree.rb

@@ -0,0 +1,123 @@
+$LOAD_PATH << "../lib"
+require "prefixtree/prefixtreenode"
+require "prefixtree/tocsv"
+
+
+
+class PrefixTree
+  attr_accessor :root, :container
+
+  def initialize
+    @root = Node.new
+    @container = []
+    @csv_container = FileWriter.new
+  end
+
+  def add(word)
+    container << word
+    current = root
+    word.each_char do |char|
+      current = adder_logic(current, char)
+    end
+    current.is_word = true
+    current
+  end
+
+  def include?(prefix)
+    current = root
+    prefix.each_char do |char|
+      return false unless current.data.include?(char)
+      current = current.data[char]
+    end
+    true
+  end
+
+  def find(prefix)
+    current = root
+    prefix.each_char do |char|
+      return false unless current.data.include?(char)
+      current = current.data[char]
+    end
+    current.is_word
+  end
+
+  def list(prefix = nil)
+    return container if prefix.nil?
+    verified_words(container, prefix)
+  end
+
+  def delete(word)
+    container.delete(word)
+    delete_recursively(word, 0, root)
+  end
+
+  def save_all_to_csv
+    @csv_container.save_to_csv(container)
+  end
+
+  def read_all_from_csv
+    @csv_container.read_from_csv
+  end
+
+  private
+
+  def adder_logic(current, char)
+    if current.data.include?(char)
+      current = current.data[char]
+    else
+      new_node = Node.new
+      current.data[char] = new_node
+      current = new_node
+    end
+    current
+  end
+
+  def verified_words(container, prefix)
+    res = []
+    container.each do |word|
+      res << word if word.start_with?(prefix)
+    end
+    res.length == 0 ? false : res
+  end
+
+  def word_exists?(word, index, current)
+    if index == word.length
+      return false unless current.is_word
+
+      current.is_word = false
+      current.data.length.zero?
+    end
+  end
+
+  def delete_reference(current, char, should_delete_ref)
+    if should_delete_ref
+      current.data.delete(char)
+      current.data.length.zero?
+    end
+  end
+
+  def delete_recursively(word, index, root)
+    current = root
+
+    word_exists?(word, index, current)
+
+    char = word[index]
+    return false unless current.data.include?(char)
+
+    next_node = current.data[char]
+    should_delete_ref = delete_recursively(word, index + 1, next_node)
+
+    delete_reference(current, char, should_delete_ref)
+    false
+  end
+end
+
+
+test = PrefixTree.new 
+
+test.add("a")
+test.save_all_to_csv
+
+
+
+

metadata

@@ -0,0 +1,45 @@
+--- !ruby/object:Gem::Specification
+name: pretrie
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- Giga Chubinidze
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2022-05-06 00:00:00.000000000 Z
+dependencies: []
+description: A data structure for easily saving and retrieving strings
+email: gchubinidze@unisens.ge
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/prefixtree.rb
+- lib/prefixtree/prefixtreenode.rb
+- lib/prefixtree/tocsv.rb
+homepage: https://rubygems.org/gems/prefixtree
+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.1.6
+signing_key: 
+specification_version: 4
+summary: Prefix tree data structure
+test_files: []