summaryrefslogtreecommitdiff
path: root/spec/leetcode/1_two_sum_spec.rb
blob: e16231723c7fa6131169d676ff24dd3f19b4dd0e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
require 'spec_helper'
description = <<-PROBLEM
Given an array of integers,
return indicies of the two numbers such that they add up to a specific target.
PROBLEM

describe description do
  def two_sum(numbers, target:)
    items = {}
    numbers.each_with_index do |number, index|
      if complement = items[target - number]
        return [index, complement].sort
      end
      items[number] = index
    end
    []
  end

  it 'returns 2 + 7' do
    numbers = [2, 7, 11, 15]
    results = two_sum(numbers, target: 9)
    expect(results).to eql([0, 1])
  end

  it 'returns 2 + 4' do
    numbers = [3, 2, 4]
    results = two_sum(numbers, target: 6)
    expect(results).to eql([1, 2])
  end

  it 'returns -3 + 3' do
    numbers = [-3, 4, 3, 90]
    results = two_sum(numbers, target: 0)
    expect(results).to eql([0, 2])
  end

  it 'returns 2 + 4' do
    numbers = [1, 2, 3, 4, 5, 6]
    results = two_sum(numbers, target: 6)
    expect(results).to eql([1, 3])
  end
end