#!/usr/bin/perl

use strict; # https://perlmonks.org/?node_id=11140759
use warnings;

use constant {
    INPUT => 0,
    TARGET => 1,
    EXPECTED => 2,
};

use Test::More;

my @tests = (
    [[2,7,11,15], 9, [0, 1]],
    [[15,11,7,2], 9, [2, 3]],
    [[-15,-11,-7,-2], -9, [2, 3]],
    [[3,2,4], 6, [1,2]],
    [[3,3], 6, [0,1]],
    [[-3,-3], -6, [0,1]],
    [[(1,6)x1995,3,5], 8, [3990,3991]], # for O(n**2) solutions
    [[-1,4, -4], 0, [1,2]],
);

plan tests => 0+@tests;

for my $test (@tests) {
    is_deeply
        two_sum($test->[INPUT], $test->[TARGET]),
        $test->[EXPECTED];
}

sub two_sum # O(n)
  {
  my ($input, $target) = @_;
  my (%h, $second);
  push @{ $h{$input->[$_]} }, $_ for 0 .. $#$input;
  $second = $h{$target - $_} and ($_ != $target - $_ or @$second > 1)
    and return [ $h{$_}[0], $second->[-1] ] for @$input;
  }

[download]

In reply to Re: LeetCode Problem #1 - Two Sum - Improve and/or discuss. by tybalt89
in thread LeetCode Problem #1 - Two Sum - Improve and/or discuss. by kcott

Title:
Use:  <p> text here (a paragraph) </p>
and:  <code> code here </code>
to format your post, it's "PerlMonks-approved HTML":


  • Posts are HTML formatted. Put <p> </p> tags around your paragraphs. Put <code> </code> tags around your code and data!
  • Titles consisting of a single word are discouraged, and in most cases are disallowed outright.
  • Read Where should I post X? if you're not absolutely sure you're posting in the right place.
  • Please read these before you post! —
  • Posts may use any of the Perl Monks Approved HTML tags:
    a, abbr, b, big, blockquote, br, caption, center, col, colgroup, dd, del, details, div, dl, dt, em, font, h1, h2, h3, h4, h5, h6, hr, i, ins, li, ol, p, pre, readmore, small, span, spoiler, strike, strong, sub, summary, sup, table, tbody, td, tfoot, th, thead, tr, tt, u, ul, wbr
  • You may need to use entities for some characters, as follows. (Exception: Within code tags, you can put the characters literally.)
            For:     Use:
    & &amp;
    < &lt;
    > &gt;
    [ &#91;
    ] &#93;
  • Link using PerlMonks shortcuts! What shortcuts can I use for linking?
  • See Writeup Formatting Tips and other pages linked from there for more info.