# # Two Sum

leetcode.com (opens new window)

## # Question

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution, and you may not use the same element twice.

Example:

``````Given nums = [2, 7, 11, 15], target = 9,

Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
``````

## # Solutions

Based on the description, we can conclude a formal description of the question, which will be the cornerstone of our solution.

``````Given nums, exist i and j: nums[i] + nums[j] = target, 0 <= i < len(nums), 0 <= j < len(nums), i ≠ j.
``````

Let's start from a brute-force algorithm.

``````def twoSum(nums, target):
for i in range(len(nums)):
for j in range(len(nums)):
if i != j and nums[i] + nums[j] == target:
return [i, j]
``````

Let's analyze this brute-force algorithm. The time complexity is O(n2) and the space complexity is O(1). Below is a short proof that it's a correct algorithm.

• The returning statement is within the context of the predicate expression, such that
• i != j, so it's under the condition of `i ≠ j`.
• nums[i] + nums[j] == target, so it's under the condition of `nums[i] + nums[j] = target`.
• Both variables `i` and `j` are from `range(len(nums))`, so they're under the condition of `0 <= i < len(nums)` and `0 <= j < len(nums)`.
• The `raise Exception` statement aborts the process and hence any other solution will not be considered.

Since the time complexity is not good at all, we now seek if it can be improved.

By moving `nums[j]` to the right side, we get:

``````nums[i] = target - nums[j]
``````

Then, by applying `f` to the both side, we get:

``````f(nums[i]) = f(target - nums[j])
``````

Here, every value of x should have corresponding value y, such that f(x) = y, and every value of y should be unique.

You might say it's hard to find such function `f`. But fortunately, there is one that you might have seen and used on a daily basis - `dict.get`, the getter function of the data structure dictionary (or HashMap, table, whatever).

We can also let `dict[num] = index`, since every index of the num in array nums is unique.

So far, we have a transformation of above formal description of the problem.

``````Given dict D, array nums,

* any j: D.get(nums[j]) = j, 0 <= j < lens(nums).
* exist i: D.get(target - nums[j]) = i, iff target - nums[j] in D.
``````

We can produce below code. Note that the first highlighted code satisfies `exist i` condition, and the second highlighted code satisfies `any j` condition.

``````def twoSum(nums, target):
D = {}
for j in range(len(nums)):
if target - nums[j] in D:
# i = D.get(nums[i]) = D.get(target - nums[j])
i = D.get(target - nums[j])
return [i, j]
else:
D[nums[j]] = j
• The returning statement is under predicate `if target - nums[j] in D:`, so it's under the condition of `iff target - nums[j] in D`.
• The assignment of variable `i` matches the existence condition of i.
• The assignment statement in the else branch establishes the truth of `any j: D.get(nums[j]) = j, 0 <= j < lens(nums)`.