About usWhy usInstructorsReviewsCostFAQContactBlogRegister for Webinar

3 Sum Problem

Given an integer array arr of size n, find all magic triplets in it.

Magic triplet is a group of three numbers whose sum is zero.

Note that magic triplets may or may not be made of consecutive numbers in arr.

Example One

Input: arr = [10, 3, -4, 1, -6, 9]

Output: [“10,-4,-6”, “3,-4,1”]

Example Two

Input: arr = [12, 34, -46]

Output: [“12,-46,34”]

Example Three

Input: arr = [0, 0, 0];

Output: [“0,0,0”]

Example Four

Input: arr = [-2, 2, 0 -2, 2];

Output: [“2,-2,0”]

Notes

Input Parameters: Function has one argument: array of integers arr.

Output: Function must return an array of strings. Each string (if any) in the array must represent a unique magic triplet and strictly follow this format: “1,2,-3” (no whitespace, one comma between numbers).

Order of the strings in the array is insignificant. Order of the integers in any string is also insignificant. For example, if [“1,2,-3”, “1,-1,0”] is a correct answer, then [“0,1,-1”, “1,-3,2”] is also a correct answer.

Triplets that only differ by order of numbers are considered duplicates, and duplicates must not be returned. For example, if “1,2,-3” is a part of an answer, then “1,-3,2”, “-3,2,1” or any permutation of the same numbers may not appear in the same answer (though any one of them may appear instead of “1,2,-3”).

Constraints:

  • 1 n
  • -1000 arr
  • arr may contain duplicate numbers.
  • arr is not necessarily sorted.

1. Brute Force Solution

A simple solution is to use three nested loops and check for each possible combination if their sum is zero. To avoid duplicate answers, we need to hash the triplet and store it in a set. An easy way to hash a triplet is by converting it to a string.

Extra space needed will be the number of unique triplets that contribute to the answer.

Space Complexity:

O(n^2)

Time Complexity:

O(n^3)


static String[] findZeroSum(int[] arr) {
        Set answer = new HashSet<>();
        // Sorting is only necessary for avoiding duplicates in the answer.
        Arrays.sort(arr);
        int n = arr.length;
        for (int index_1 = 0; index_1 < n; index_1++) {
            for (int index_2 = index_1 + 1; index_2 < n; index_2++) {
                for (int index_3 = index_2 + 1; index_3 < n; index_3++) {
                    int sum = arr[index_1] + arr[index_2] + arr[index_3];
                    if (sum == 0) {
                        answer.add(arr[index_1] + "," + arr[index_2] + "," + arr[index_3]);
                    }
                }
            }
        }
        return answer.toArray(new String[0]);
    }
    

2. Optimal Solution

Optimal solution uses the two pointer method. We maintain left and right pointers to elements of a sorted array. If their sum is greater than intended we decrease the right pointer, otherwise we increase the left pointer. This method works in linear time (and to solve this particular problem we use this algorithm n times, once for each element in arr).

First, we will sort arr (this will contribute O(n*log(n)) to the time complexity). Then, for every element or arr, we will apply the two pointer technique to find any magic triplets that include that element. We will then add unique triplets to the answer.

Time Complexity:

O(n^2 + n*log(n))

Space Complexity:

O(n^2)



static String[] findZeroSum(int[] arr) {
        Set answer = new HashSet<>();
        Arrays.sort(arr); // A prerequisite for the two pointer technique to work.
        int n = arr.length;
        for (int index = 0; index < n; index++) {
            int currentElement = arr[index];
            // We will look for two elements that sum up to this:
            int neededSum = -currentElement;
            int left = index + 1, right = n - 1;
            while (left < right) {
                int sum = arr[left] + arr[right];
                if (sum == neededSum) {
                    answer.add(currentElement + "," + arr[left] + "," + arr[right]);
                    left++; // "right--" would also work fine here.

                    // Note that the three numbers in our magic triplet strings
                    // will always be sorted in the increasing order because
                    // we sorted the array and because index > left > right.
                    // That and using a set to store the strings is enough for
                    // avoiding duplicates in the answer.
                } else if (sum > neededSum) {
                    right--;
                } else {
                    left++;
                }
            }
        }
        return answer.toArray(new String[0]);
    }
    


Attend our Free Webinar on How to Nail Your Next Technical Interview

Recommended Posts

All Posts