The naive solution for this problem is pretty trivial. Brute force every pair-sum and check if the sum is K. You could be a little smart about this. You initially sort the array O(nlogn) and iterate the array pairwise and quit when the sum exceeds K, you can quit the inner loop since the array is sorted the sum can only increase.
Problem : Given an array with distinct elements. Find out if there exists an instance where sum of two distinct elements is equal to a given constant K. Say, A = [1,2,3,4], K = 7. Output should be 3,4
For the rest of the article, I'll be assuming K = 0, this doesn't affect the solutions in any way and it's trivial to introduce a new variable into the code without major changes.
Here's the code, should be easy to follow... I'll be using Java today.
Now that the naive solution is out of the way, let's try and do some clever stuff. So notice we can afford to sort the array since the brute solution was already O(n^2) (which is the higher order term). Now a very big advantage of a sorted array is we can search it in O(logn), so instead of the inner loop doing a linear search for (K-A[i]), we perform a Binary Search. The solution now drops to O(nlogn).
The Code as follows.
We maintain two indices left and right. The left one points to the beginning of the array (ie the minimum element) and the right index is equal to the last element (ie the maximum element). Now we calculate S where S = A[left] + A[right]. There are 3 cases.
- S == K, which is the best case scenario. We found the pair (A[left], A[right]) whose sum = K
- S > K, Now if the sum is greater than K, then it means the right element is more than it should be. We want to reduce S, if we increment left, S will increase which doesn't help. If we decrement right, the sum will decrease. Note, we don't decrement left, the part of the array before left has already been processed to not contain the solution.
- If S < K, as you can imagine, since S is lower than the required sum, we need to "add" more value to it, hence we increment left.
The process repeats till left < right, at which point you can conclude there is no solution.
Let's take a short example, say the array is [-10,-5,-2,0,1,2,3], K = 0
- left = 0, right = 6, A[left] = -10, A[right] = 3, S = -7, Since S < 0, we increment left
- left = 1, right = 6, A[left] = -5, A[right] = 3, S= -2, Since S < 0, we increment left
- left = 2, right = 6, A[left] = -2, A[right] = 3, S = 1, Since S > 0, we decrement right
- left = 2, right = 5, A[left] = -2, A[right] = 2, S = 0, we found the solution.
Hopefully, that helps. Here's the code that implements the above. Again, this only works with a sorted array. The Complexity is Linear since at each iteration one element is eliminated.
Comments
Post a Comment