454. 4Sum II

MediumArrayHash Table
Leetcode Link

Problem Description

The problem provides us with four integer arrays nums1, nums2, nums3, and nums4, each of the same length n. Your task is to find the number of quadruplets (i, j, k, l) such that the sum of the elements at these indices from each array equals zero, that is nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0. The indices i, j, k, l vary from 0 to n-1.

Intuition

To find quadruplets that sum up to zero, a naive approach would involve checking all possible combinations of indices (i, j, k, l) from the four arrays, which leads to a solution with a time complexity of O(n^4). However, this isn't efficient when n is large.

Instead, we can first consider pairs of elements from nums1 and nums2. We record the possible sums that these pairs can make and the frequency of each sum. A Counter in Python can be used effectively for this purpose.

Then, we can look for the pairs of elements from nums3 and nums4 that, when added together, create a sum that negates the sum we obtained from nums1 and nums2. This way, the total sum will be zero. We find these complementary sums by iterating through all combinations of nums3 and nums4 and checking if the negated sum is present in our Counter (which is populated by sums of nums1 and nums2).

Using this method significantly reduces the time complexity because we only consider two loops of n^2 operations each, hence O(n^2). Furthermore, since we are storing the sum and frequency in a Counter (which is a hashmap, effectively), our lookups for the complementary sum are done in constant time.

Solution Approach

The implementation of the solution uses a two-step process that takes advantage of the Counter class in Python, which is a type of dictionary designed for counting hashable objects, a subclass of dict. Here's a breakdown of the algorithm:

Step 1: Compute Pair Sums from nums1 and nums2

Firstly, we iterate over all pairs of elements (a, b) where a is from nums1 and b is from nums2. We calculate their sum and store this in a Counter dictionary, which will hold the sum as keys and the frequency of these sums as values. This dictionary will capture all possible pairwise sums along with how many times each sum occurs.

cnt = Counter(a + b for a in nums1 for b in nums2)

Step 2: Find Complementary Sums from nums3 and nums4

Next, for each combination of elements (c, d) from nums3 and nums4, we calculate their sum and look for the complement of this sum in our previously populated Counter (specifically, we're looking for -(c + d)). If found, it means there exists at least one pair from nums1 and nums2 that can be added to this nums3 and nums4 pair to make the total sum zero. The frequency stored in the Counter for -(c + d) represents how many such pairs we have from nums1 and nums2.

return sum(cnt[-(c + d)] for c in nums3 for d in nums4)

By iterating over all pairs from nums3 and nums4 and summing up the counts from the Counter, we get the total number of valid quadruplets that meet our condition.

This approach effectively reduces a complex problem into a simpler one that requires fewer computations by dividing it into two parts. It utilizes the power of hashmaps to speed up lookups for complementary pairs, ensuring an overall time complexity of O(n^2) which is a significant improvement over the brute force approach with O(n^4). Also, because we are storing up to n^2 pairs in the Counter, the space complexity is also O(n^2).

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
        cnt = Counter(a + b for a in nums1 for b in nums2)
        return sum(cnt[-(c + d)] for c in nums3 for d in nums4)

Notice how the implementation is cogent and does not require nested loops over all four arrays, significantly enhancing efficiency.

Ready to land your dream job?

Unlock your dream job with a 2-minute evaluator for a personalized learning plan!

Start Evaluator

Example Walkthrough

Let's go through a small example using the provided solution approach to understand how it works in practice. We will take four arrays with a smaller length for simplicity.

Suppose nums1 = [1, -1], nums2 = [-1, 1], nums3 = [0, 1], and nums4 = [1, -1], and we need to find the number of quadruplets such that nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0.

Step 1: Compute Pair Sums from nums1 and nums2

Firstly, we calculate all possible sums of pairs (a, b) where a is from nums1 and b is from nums2, and store these sums in the Counter:

  • Pair (1, -1): sum is 0
  • Pair (1, 1): sum is 2
  • Pair (-1, -1): sum is -2
  • Pair (-1, 1): sum is 0

Thus, the Counter after step 1 will look like this: {0: 2, 2: 1, -2: 1}, which tells us that the sum 0 appears twice and the sums 2 and -2 appear once.

Step 2: Find Complementary Sums from nums3 and nums4

Now, we need to find pairs (c, d) where c is from nums3 and d is from nums4 such that - (c + d) is in the Counter.

  • Pair (0, 1): sum is 1 and its complement -1 is not in the Counter, so this pair does not contribute.
  • Pair (0, -1): sum is -1 and its complement 1 is not in the Counter, so no contribution from this pair either.
  • Pair (1, 1): sum is 2 and its complement -2 is in the Counter and it appears once, so we have one quadruplet: (1,1,1,1).
  • Pair (1, -1): sum is 0 and its complement 0 is in the Counter appearing twice, contributing two quadruplets: (1,-1,1,-1) and (-1,1,1,-1).

By iterating over the last two arrays and summing up the counts for each valid complement in the Counter, we find a total of 3 quadruplets.

Hence, for the given small example, the fourSumCount method would return 3. This illustrates that even for small arrays, the algorithm effectively combines pairs from the first two and last two arrays, utilizing the Counter to manage and efficiently count the complementary pairs, achieving a significant performance gain compared to the brute force approach.

Solution Implementation

1from collections import Counter
2from typing import List
3
4class Solution:
5    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
6        # Create a Counter to store the frequency of the sums of pairs taken from nums1 and nums2
7        pairwise_sum_count = Counter(a + b for a in nums1 for b in nums2)
8
9        # Accumulate the number of times the current sum of pairs from nums3 and nums4
10        # when added to the pair sums from nums1 and nums2 gives zero (i.e., sum to zero).
11        count = 0
12        for c in nums3:
13            for d in nums4:
14                # The target is the negative of the sum of c and d which would give zero when added to a pair sum from nums1 and nums2.
15                target = -(c + d)
16                # If the target exists in the Counter, add the frequency to the count
17                count += pairwise_sum_count[target]
18
19        return count
20
1class Solution {
2
3    // This method finds the count of tuples (i, j, k, l) such that nums1[i] + nums2[j] + nums3[k] + nums4[l] is zero.
4    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
5        // HashMap to store the frequency of the sum of elements from nums1 and nums2
6        Map<Integer, Integer> sumFrequencyMap = new HashMap<>();
7      
8        // Calculate all possible sums of pairs from nums1 and nums2 and store frequencies in the map
9        for (int num1 : nums1) {
10            for (int num2 : nums2) {
11                sumFrequencyMap.merge(num1 + num2, 1, Integer::sum);
12            }
13        }
14      
15        // Initialize the count of valid tuples to 0
16        int countOfValidTuples = 0;
17      
18        // For each possible pair of nums3 and nums4, check if the negative sum already exists in our map
19        for (int num3 : nums3) {
20            for (int num4 : nums4) {
21                // The count is incremented by the frequency of the negated sum, 
22                // if it exists, indicating valid tuples that add up to zero
23                countOfValidTuples += sumFrequencyMap.getOrDefault(-(num3 + num4), 0);
24            }
25        }
26      
27        // Return the total count of valid tuples
28        return countOfValidTuples;
29    }
30}
31
1#include <vector>
2#include <unordered_map>
3using namespace std;
4
5class Solution {
6public:
7    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
8        // Create a hash map to store the frequency of the sum of pairs from nums1 and nums2
9        unordered_map<int, int> sumCount;
10      
11        // Calculate all possible sums of pairs from nums1 and nums2, and record the frequency
12        for (int num1 : nums1) {
13            for (int num2 : nums2) {
14                sumCount[num1 + num2]++;
15            }
16        }
17      
18        // Initialize the answer to 0. This will hold the number of tuples such that the sum is 0
19        int count = 0;
20      
21        // For every pair from nums3 and nums4, check if the opposite number exist in the sumCount map
22        for (int num3 : nums3) {
23            for (int num4 : nums4) {
24                // Find the complement of the current sum in the hash map
25                auto it = sumCount.find(-(num3 + num4));
26              
27                // If the complement is found, this means there are tuples from nums1 and nums2
28                // that, when added with the current pair from nums3 and num4, sum to 0
29                if (it != sumCount.end()) {
30                    // Add the frequency of the complement to the count
31                    count += it->second;
32                }
33            }
34        }
35      
36        // Return the total count of tuples resulting in a sum of 0
37        return count;
38    }
39};
40
1// Function to count the number of tuples (a, b, c, d) from four lists
2// such that a + b + c + d is equal to 0.
3function fourSumCount(nums1: number[], nums2: number[], nums3: number[], nums4: number[]): number {
4    // Create a map to store the frequency of sums of pairs from the first two lists.
5    const sumFrequency: Map<number, number> = new Map();
6  
7    // Calculate all possible sums from nums1 and nums2 and update the frequency map.
8    for (const num1 of nums1) {
9        for (const num2 of nums2) {
10            const sum = num1 + num2;
11            sumFrequency.set(sum, (sumFrequency.get(sum) || 0) + 1);
12        }
13    }
14  
15    // Initialize a variable to keep track of the number of valid tuples found.
16    let tupleCount = 0;
17  
18    // For each possible sum of pairs from nums3 and nums4, check if the negation
19    // of the sum is present in the frequency map. If so, increase the count of valid tuples.
20    for (const num3 of nums3) {
21        for (const num4 of nums4) {
22            const sum = num3 + num4;
23            tupleCount += sumFrequency.get(-sum) || 0;
24        }
25    }
26  
27    // Return the total count of valid tuples.
28    return tupleCount;
29}
30

Time and Space Complexity

The given Python code is designed to find the number of tuples (i, j, k, l) such that nums1[i] + nums2[j] + nums3[k] + nums4[l] is zero.

Time Complexity:

The time complexity is analyzed based on the number of operations performed by the code:

  • The first part of the code creates a Counter object to count the frequency of sums of pairs taken from nums1 and nums2. This involves iterating over each element in nums1 and nums2, which, if the length of the lists is n, results in n * n or n^2 operations.

  • The second part computes the sum of frequencies of the complement of each possible sum in nums3 and nums4 that makes the total sum zero. This also requires n * n or n^2 operations.

Combining these two, we get a total of 2 * n^2 operations, but since constants are neglected in Big O notation, the time complexity of the code is O(n^2).

Space Complexity:

The space complexity is considered based on the additional memory used by the program:

  • The Counter object holds at most n^2 entries, corresponding to each unique sum of elements from nums1 and nums2.

  • No other significant extra space is used for computation since the second sum is computed iteratively and sums are not stored.

Hence, the space complexity of the code is O(n^2).

Learn more about how to find time and space complexity quickly using problem constraints.


Discover Your Strengths and Weaknesses: Take Our 2-Minute Quiz to Tailor Your Study Plan:
Question 1 out of 10

What does the following code do?

1def f(arr1, arr2):
2  i, j = 0, 0
3  new_arr = []
4  while i < len(arr1) and j < len(arr2):
5      if arr1[i] < arr2[j]:
6          new_arr.append(arr1[i])
7          i += 1
8      else:
9          new_arr.append(arr2[j])
10          j += 1
11  new_arr.extend(arr1[i:])
12  new_arr.extend(arr2[j:])
13  return new_arr
14
1public static List<Integer> f(int[] arr1, int[] arr2) {
2  int i = 0, j = 0;
3  List<Integer> newArr = new ArrayList<>();
4
5  while (i < arr1.length && j < arr2.length) {
6      if (arr1[i] < arr2[j]) {
7          newArr.add(arr1[i]);
8          i++;
9      } else {
10          newArr.add(arr2[j]);
11          j++;
12      }
13  }
14
15  while (i < arr1.length) {
16      newArr.add(arr1[i]);
17      i++;
18  }
19
20  while (j < arr2.length) {
21      newArr.add(arr2[j]);
22      j++;
23  }
24
25  return newArr;
26}
27
1function f(arr1, arr2) {
2  let i = 0, j = 0;
3  let newArr = [];
4  
5  while (i < arr1.length && j < arr2.length) {
6      if (arr1[i] < arr2[j]) {
7          newArr.push(arr1[i]);
8          i++;
9      } else {
10          newArr.push(arr2[j]);
11          j++;
12      }
13  }
14  
15  while (i < arr1.length) {
16      newArr.push(arr1[i]);
17      i++;
18  }
19  
20  while (j < arr2.length) {
21      newArr.push(arr2[j]);
22      j++;
23  }
24  
25  return newArr;
26}
27

Recommended Readings

Want a Structured Path to Master System Design Too? Donā€™t Miss This!


Load More