python_tasks/Two Sum.py at main · Krupshark/python_tasks · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
"""
Task:

Write a function that takes an array of numbers (integers for the tests) and a target number.
It should find two different items in the array that, when added together, give the target value.
The indices of these items should then be returned in a tuple / list (depending on your language) like so: (index1, index2).

For the purposes of this kata, some tests may have multiple answers; any valid solutions will be accepted.

The input will always be valid (numbers will be an array of length 2 or greater, and all of the items will be numbers;
target will always be the sum of two different items from that array).

Solution:
"""


def two_sum(numbers, target):
    for i, number_a in enumerate(numbers):
        for j, number_b in enumerate(numbers):
            if number_a + number_b == target and i != j:
                return (i, j)
    return False