rb_binary_heap 2.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 2a3883375b2b15e01fdb4876a3e19143424ce70397ffb588f009cdc3fd94bfee
+  data.tar.gz: 76e7611f011248d9e6cebe35d6bce18e416ec2df52d0efe33a29269598cbbbf7
+SHA512:
+  metadata.gz: 2900b3619db8f8a1b5a109b6cb882c1b7f51bb963f8502a547711edc598fffaa7628f4f1b26e11b808747ab2ab1509127780606947f1a253687776589b980a46
+  data.tar.gz: '04193409cf464213dae17f7e3b9fa6027887a7b6c1ebb59ca065ed71db79f8576b1a8ebe69fc994e94bc97341905da020eb7fae72918d5ec1cfb5a7f4b51e02f'

data/lib/rb_binary_heap.rb

@@ -0,0 +1,142 @@
+# frozen_string_literal: true
+
+class BinaryHeap
+  include Enumerable
+
+  def initialize(array = [], type: nil, compare_attribute: nil, &block)
+    @array = array
+
+    comparison_method_name = :compare
+
+    if block_given?
+      if type || compare_attribute
+        raise 'Heap: :type and :compare_attribute should not be defined if the block is given'
+      end
+
+      define_singleton_method(comparison_method_name) do |a, b|
+        block.call a, b
+      end
+    else
+      type = :min if type.nil?
+
+      op = heap_type_to_comparison_operator type
+
+      if compare_attribute.nil?
+        define_singleton_method(comparison_method_name) do |a, b|
+          a.send(op, b)
+        end
+      else
+        define_singleton_method(comparison_method_name) do |a, b|
+          a_value = a.send(compare_attribute)
+          b_value = b.send(compare_attribute)
+          a_value.send(op, b_value)
+        end
+      end
+    end
+
+    heapify
+  end
+
+  def top
+    raise "Heap: Can't get the top element, heap is empty" if @array.empty?
+    @array.first
+  end
+
+  def push(value)
+    @array.push value
+    sift_up_from_last
+    self
+  end
+
+  def pop
+    raise "Heap: Can't pop, heap is empty" if @array.empty?
+    return @array.pop if @array.size == 1
+
+    result = @array.first
+    last_element = @array.pop
+    @array[0] = last_element
+    
+    sift_down_from_first
+    result
+  end
+
+  def not_empty?
+    !empty?
+  end
+
+  def empty?
+    @array.empty?
+  end
+
+  def each
+    yield pop while not_empty?
+  end
+
+  def size
+    @array.size
+  end
+
+  private
+
+  def heapify
+    i = @array.size / 2
+    while i >= 0
+      sift_down_from i
+      i -= 1
+    end
+  end
+
+  def sift_up_from_last
+    sift_up_from @array.size - 1
+  end
+
+  def sift_down_from_first
+    sift_down_from 0
+  end
+
+  def sift_up_from(i)
+    while i.positive?
+      parent = (i - 1) / 2
+
+      break if compare(@array[parent], @array[i])
+
+      swap i, parent
+      
+      i = parent
+    end
+  end
+
+  def sift_down_from(i)
+    while (left = 2 * i + 1) < @array.size
+      largest = i
+
+      largest = left if !compare(@array[largest], @array[left])
+
+      right = 2 * i + 2
+
+      if right < @array.size && !compare(@array[largest], @array[right])
+        largest = right
+      end
+
+      break if largest == i
+
+      swap i, largest
+
+      i = largest
+    end
+  end
+
+  def swap(i, j)
+    @array[i], @array[j] = @array[j], @array[i]
+  end
+
+  def heap_type_to_comparison_operator(type)
+    case type
+    when :min then :<
+    when :max then :>
+    else
+      raise 'Heap: :type should be either :min or :max'
+    end
+  end
+end
+

metadata

@@ -0,0 +1,43 @@
+--- !ruby/object:Gem::Specification
+name: rb_binary_heap
+version: !ruby/object:Gem::Version
+  version: '2.1'
+platform: ruby
+authors:
+- metacircu1ar
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2023-01-20 00:00:00.000000000 Z
+dependencies: []
+description: Implementation of a binary heap in Ruby. See github for usage examples.
+email: ''
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/rb_binary_heap.rb
+homepage: https://github.com/metacircu1ar/rb_binary_heap
+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.4.1
+signing_key: 
+specification_version: 4
+summary: Binary heap
+test_files: []