0% completed
Problem Statement
Given an array of numbers sorted in ascending order and a target sum, find a pair in the array whose sum is equal to the given target. If no such pair exists return [-1, -1].
Write a function to return the indices of the two numbers (i.e. the pair) such that they add up to the given target.
Example 1:
Input: [1, 2, 3, 4, 6], target=6
Output: [1, 3]
Explanation: The numbers at index 1 and 3 add up to 6: 2+4=6
Example 2:
Input: [2, 5, 9, 11], target=11
Output: [0, 2]
Explanation: The numbers at index 0 and 2 add up to 11: 2+9=11
Constraints:
- 2 <= arr.length <= 10<sup>4</sup>
- -10<sup>9</sup> <= arr[i] <= 10<sup>9</sup>
- -10<sup>9</sup> <= target <= 10<sup>9</sup>
Solution
Since the given array is sorted, a brute-force solution could be to iterate through the array, taking one number at a time and searching for the second number through Binary Search. The time complexity of this algorithm will be O(N*logN). Can we do better than this?
To solve this problem, we can use a two-pointer approach. This approach is efficient because it takes advantage of the sorted nature of the array. By starting with one pointer at the beginning and the other at the end, we can adjust their positions based on the sum of the elements they point to. This allows us to find the pair that adds up to the target without needing to check all possible pairs, which saves time.
By moving the pointers inward, we can systematically find the pair in a single pass through the array. This ensures that the solution is both time-efficient and easy to implement.
Step-by-Step Algorithm
- Initialize two pointers: Start with one pointer (
left) at the beginning (index 0) and the other pointer (right) at the end (last index) of the array. - Loop until pointers meet: Continue the loop until
leftis less thanright.- Calculate current sum: Add the elements at the
leftandrightpointers. - Check if the sum matches the target:
- If
currentSumequals the target sum, return the indices[left, right]. - If
currentSumis less than the target sum, increment theleftpointer to increase the sum. - If
currentSumis more than the target sum, decrement therightpointer to decrease the sum.
- If
- Calculate current sum: Add the elements at the
- Return default values: If no pair is found, return
[-1, -1].
Algorithm Walkthrough
Let's walk through the example with input [1, 2, 3, 4, 6] and target 6.
- Initialize pointers:
left = 0,right = 4 - First iteration:
currentSum = 1 + 6 = 7(greater than target)- Decrement
rightto3
- Second iteration:
currentSum = 1 + 4 = 5(less than target)- Increment
leftto1
- Third iteration:
currentSum = 2 + 4 = 6(equals target)- Return indices
[1, 3]
Here is the visual representation of this algorithm for Example-1:
Code
Here is what our algorithm will look like:
Time Complexity
-
Initialization: Constant time, O(1), as it involves assigning values to
leftandright. -
While Loop:
- The
whileloop runs as long asleftis less thanright. - In the worst case, this loop iterates over all elements of the array once. This happens when no pair is found, or the pair is found at the extreme ends of the array.
- Each iteration involves a constant amount of work: calculating
currentSum, comparing it withtargetSum, and then incrementingleftor decrementingright.
Therefore, the loop runs in O(n) time, where n is the number of elements in the array.
- The
-
Overall: The dominant factor in this algorithm is the while loop, making the overall time complexity O(n).
Space Complexity
- The algorithm uses a fixed amount of extra space: variables
left,right, andcurrentSum. - It does not depend on the size of the input array, as no additional data structures are used that grow with the input size.
- Thus, the space complexity is O(1), constant space.
In summary, the algorithm has a time complexity of O(n) and a space complexity of O(1).
An Alternate approach
Instead of using a two-pointer or a binary search approach, we can utilize a HashTable to search for the required pair. We can iterate through the array one number at a time. Let’s say during our iteration we are at number ‘X’, so we need to find ‘Y’ such that “X+Y==Target”. We will do two things here:
Search for ‘Y’ (which is equivalent to “Target−X”) in the HashTable. If it is there, we have found the required pair. Otherwise, insert “X” in the HashTable, so that we can search it for the later numbers.
Step-by-step Algorithm
-
Initialize a HashMap:
- Create a HashMap to store numbers as keys and their indices as values.
-
Iterate through the array:
- Loop through each element in the array using a for loop.
-
Check for the complement:
- For each element, check if the HashMap contains the complement (i.e.,
targetSum - current element). - If it does, return the indices of the complement and the current element.
- For each element, check if the HashMap contains the complement (i.e.,
-
Store the element and its index:
- If the complement is not found, store the current element and its index in the HashMap.
-
Return result:
- If no pair is found by the end of the loop, return
[-1, -1].
- If no pair is found by the end of the loop, return
Algorithm Walkthrough
Example Input: [1, 2, 3, 4, 6], Target: 6
-
Initialize:
nums = {}
-
First iteration (i = 0):
- Current element:
1 - Complement needed:
6 - 1 = 5 5is not innums- Store
1with index0:nums = {1: 0}
- Current element:
-
Second iteration (i = 1):
- Current element:
2 - Complement needed:
6 - 2 = 4 4is not innums- Store
2with index1:nums = {1: 0, 2: 1}
- Current element:
-
Third iteration (i = 2):
- Current element:
3 - Complement needed:
6 - 3 = 3 3is not innums- Store
3with index2:nums = {1: 0, 2: 1, 3: 2}
- Current element:
-
Fourth iteration (i = 3):
- Current element:
4 - Complement needed:
6 - 4 = 2 2is innumswith index1- Return indices
[1, 3]
- Current element:
Code
Time Complexity
-
HashMap Initialization: Constant time, O(1).
-
For Loop:
- The
forloop iterates over each element of the array once. - In each iteration, the algorithm checks if the element is already present in the
HashMap(ordictionaryin Python) and either returns a result or inserts an element into theHashMap. - This element checking or insertion operations of a
HashMapgenerally operate in O(1) time due to efficient hashing. However, in the worst case (e.g., when many hash collisions occur), these operations can degrade to O(n). Under the assumption of a good hash function with minimal collisions, these operations can be considered O(1).
Therefore, the loop runs in O(n) time in the average case, where (n) is the number of elements in the array.
- The
-
Overall: The dominating factor in this algorithm is the for loop. Under the assumption of efficient hashing, the overall average time complexity is O(n).
Space Complexity
- The algorithm uses a
HashMapto store elements from the array. In the worst case, this map can store all elements of the array if no pair is found that adds up to the target sum - Thus, the space complexity is proportional to the number of elements in the array, which is O(n).
In summary, the algorithm has an average time complexity of O(n) and a space complexity of O(n). The time complexity can degrade to O(n^2) in the worst case due to potential hash collisions, but this is generally not the common case with a good hash function.
On this page
Problem Statement
Solution
Step-by-Step Algorithm
Algorithm Walkthrough
Code
Time Complexity
Space Complexity
An Alternate approach
Step-by-step Algorithm
Algorithm Walkthrough
Code
Time Complexity
Space Complexity