ruby-heap 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b49b7d675703a399ead164dfd7cd837975253c61
-  data.tar.gz: ac4a978badfa483cadb5445bdf308283af462ee7
+  metadata.gz: 67185670438a9077e5eec874db91d471ce687900
+  data.tar.gz: efbfb03e0365fbff555a2e94b7d2c6873a6d3515
 SHA512:
-  metadata.gz: 604bab2bb181daf0cbc411c5fae7788ab5424bd517497f1f6bb1b80bea7473eafe72f0ddfa4d0dc56cbc6ea7feecb73eedede1a12e4c0b6f488ca6e3057ee7bd
-  data.tar.gz: 0e725440379858b4f3dbe03c0686b36c800775ab3ad5a33726edff9aa6ef30fb316c1f4aaacf11c649a5dca78a06c9cf07f84143801dece3a2064e1a8eb98b9e
+  metadata.gz: 842e05fb7274881285c6a090c1e4abf03275c78544d9d52515d13e0a6e7eddc8b0d9622228b9f184a136668aaff5e8b14196912554979ef0f9579d09946b7ad1
+  data.tar.gz: 11d8a860de8756728694120d5d3bdf53c232764656505a929a7f74514b09c879242733fcd534ef2cd709cd3e7fd1809f8a01ee5614d4e6d1fb1515896b5ebafb

data/lib/Heap.rb

@@ -1,5 +1,6 @@
 require 'Heap/version'
 
 module Heap
-  require 'Heap/binary_heap/binary_heap'
+  require 'Heap/binary_heap/binary_heap_min'
+  require 'Heap/binary_heap/binary_heap_max'
 end

data/lib/Heap/binary_heap/binary_heap_max.rb

@@ -0,0 +1,85 @@
+module Heap
+  module BinaryHeap
+    class MaxHeap
+      attr_reader :elements
+
+      def initialize(elements = [])
+        @elements = []
+        add(elements.pop) until elements.empty?
+      end
+
+      def add(element)
+        @elements.push element
+        swim_up(count)
+      end
+
+      def count
+        @elements.length
+      end
+
+      def extract_max
+        @elements[0]
+      end
+
+      def extract_max!
+        swap(1, count)
+        el = @elements.pop
+        swim_down(1)
+        el
+      end
+
+      def sort
+        el_temp = @elements.clone
+        result = []
+        result.push extract_max! while count > 0
+        @elements = el_temp
+        result
+      end
+
+      private
+
+      def swap(index1, index2)
+        temp = @elements[index1 - 1]
+        @elements[index1 - 1] = @elements[index2 - 1]
+        @elements[index2 - 1] = temp
+      end
+
+      def swim_up(index)
+        return if index == 1
+        parent_index = (index / 2).floor
+        return if @elements[parent_index - 1] >= @elements[index - 1]
+        swap(parent_index, index)
+        swim_up parent_index
+      end
+
+      def swim_down(index)
+        child1_index = 2 * index
+        child2_index = 2 * index + 1
+        return if @elements[child1_index - 1].nil? && @elements[child2_index - 1].nil?
+        if @elements[child2_index - 1].nil?
+          return if @elements[child1_index - 1] <= @elements[index - 1]
+          swap(index, child1_index)
+          swim_down(child1_index)
+        else
+          if @elements[child2_index - 1] <= @elements[index - 1] && @elements[child1_index - 1] <= @elements[index - 1]
+            return
+          elsif @elements[child2_index - 1] > @elements[index - 1] && @elements[child1_index - 1] > @elements[index - 1]
+            if @elements[child2_index - 1] > @elements[child1_index - 1]
+              swap(child2_index, index)
+              swim_down(child2_index)
+            else
+              swap(child1_index, index)
+              swim_down(child1_index)
+            end
+          elsif @elements[child2_index - 1] > @elements[index - 1]
+            swap(child2_index, index)
+            swim_down(child2_index)
+          elsif @elements[child1_index - 1] > @elements[index - 1]
+            swap(child1_index, index)
+            swim_down(child1_index)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/Heap/version.rb

@@ -1,3 +1,3 @@
 module Heap
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-heap
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Alexandr Sysoev
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-06-07 00:00:00.000000000 Z
+date: 2017-06-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -69,9 +69,10 @@ files:
 - bin/console
 - bin/setup
 - lib/Heap.rb
-- lib/Heap/binary_heap/binary_heap.rb
+- lib/Heap/binary_heap/binary_heap_max.rb
+- lib/Heap/binary_heap/binary_heap_min.rb
 - lib/Heap/version.rb
-- ruby-heap-0.0.5.gem
+- ruby-heap-0.1.0.gem
 homepage: https://github.com/pups3s/ruby-heap
 licenses:
 - MIT