add reject and reject! to SortedArray

lib/sorted_containers/sorted_array.rb

@@ -1119,6 +1119,39 @@ def rassoc(obj)
       index.nil? ? nil : self[index]
     end
 
+    # Returns a new SortedArray whose elements are all those from +self+ for which the block returns false or nil.
+    #
+    # Returns an Enumerator if no block is given.
+    #
+    # @yield [value] The block to reject with.
+    # @return [SortedArray, Enumerator] The rejected array.
+    def reject
+      return to_enum(:reject) unless block_given?
+
+      select { |value| !yield(value) }
+    end
+
+    # Deletes every element of +self+ for which block evaluates to true.
+    #
+    # Returns +self+ if any changes were made, otherwise returns +nil+.
+    #
+    # Returns an Enumerator if no block is given.
+    #
+    # @yield [value] The block to reject with.
+    # @return [SortedArray, Enumerator] The rejected array.
+    def reject!
+      return to_enum(:reject!) unless block_given?
+
+      indexes_to_delete = []
+      each_with_index do |value, index|
+        indexes_to_delete << index if yield(value)
+      end
+      return nil if indexes_to_delete.empty?
+
+      indexes_to_delete.reverse.each { |index| delete_at(index) }
+      self
+    end
+
     private
 
     # Performs a left bisect on the array.

spec/sorted_array_spec.rb

@@ -1747,6 +1747,47 @@ def dig(key)
     end
   end
 
+  describe "reject" do
+    it "should reject elements that meet the given criterion" do
+      array = SortedContainers::SortedArray.new([1, 2, 3, 4, 5])
+      expect(array.reject { |i| i > 3 }).to eq(SortedContainers::SortedArray.new([1, 2, 3]))
+    end
+
+    it "should not modify the original array" do
+      array = SortedContainers::SortedArray.new([1, 2, 3, 4, 5])
+      array.reject { |i| i > 3 }
+      expect(array).to eq(SortedContainers::SortedArray.new([1, 2, 3, 4, 5]))
+    end
+
+    it "should return an enumerator if no block is given" do
+      array = SortedContainers::SortedArray.new([1, 2, 3, 4, 5])
+      expect(array.reject).to be_a(Enumerator)
+    end
+  end
+
+  describe "reject!" do
+    it "should reject elements that meet the given criterion" do
+      array = SortedContainers::SortedArray.new([1, 2, 3, 4, 5])
+      array.reject! { |i| i > 3 }
+      expect(array).to eq(SortedContainers::SortedArray.new([1, 2, 3]))
+    end
+
+    it "should return self if elements are rejected" do
+      array = SortedContainers::SortedArray.new([1, 2, 3, 4, 5])
+      expect(array.reject! { |i| i > 3 }).to eq(array)
+    end
+
+    it "should return an enumerator if no block is given" do
+      array = SortedContainers::SortedArray.new([1, 2, 3, 4, 5])
+      expect(array.reject!).to be_a(Enumerator)
+    end
+
+    it "should return nil if no elements meet the given criterion" do
+      array = SortedContainers::SortedArray.new([1, 2, 3, 4, 5])
+      expect(array.reject! { |i| i > 5 }).to be_nil
+    end
+  end
+
   describe "shift" do
     it "should shift the first value from the array" do
       array = SortedContainers::SortedArray.new([1, 2, 3, 4, 5])