1213. Intersection of Three Sorted Arrays

EasyArrayHash TableBinary SearchCounting

Problem Description

The goal is to identify common integers that are present in three different sorted arrays arr1, arr2, and arr3. These arrays are given to be sorted in strictly increasing order, which means there are no duplicate elements within each array and each subsequent element is greater than the previous one. We need to find a list of integers that are present in all three arrays and return this list in sorted order. Since the arrays are already sorted, the resulting list of common elements will also be automatically sorted.

Intuition

To find the common elements across the three sorted arrays, we could consider each element in the first array arr1 and check if it is present in both arr2 and arr3. This approach would take O(n1 * (n2 + n3)) time, where n1, n2, and n3 are the lengths of arr1, arr2, and arr3 respectively, because we would search for each element of arr1 in both arr2 and arr3.

A more efficient way to solve this would be to use the three-pointer technique, where we can iterate through all three arrays simultaneously to find common elements. This approach takes advantage of the fact that the arrays are sorted. However, the reference solution approach suggests "Binary search".

Binding together the sorted arrays property and the proposed "Binary search" approach would involve iterating through one array, and for each element in that array, perform binary search in the other two arrays. This efficiently checks for the presence of the element in the other arrays since the arrays are sorted. Binary search greatly reduces the time complexity to O(n1 * log(n2) * log(n3)).

The solution given in the code seems to use a different approach: a frequency counter. By concatenating the arrays and counting the frequencies of the elements, we can simply look for those numbers in arr1 (or any of the three arrays) that have a frequency of 3. This works under the assumption that the input arrays are strictly increasing and therefore there are no duplicates within the same array. Hence, an element with a count of 3 must appear once in each of the arrays. This approach has a time complexity of O(n1 + n2 + n3), the time it takes to combine the arrays and count the elements.

In summary, while the reference suggests to use binary search to enhance the searching step by capitalizing on the sorted property of arrays, the implemented solution uses a frequency counter for its simplicity and because it also leads to an efficient solution given the constraints of the problem.

Learn more about Binary Search patterns.

Solution Approach

The solution provided in the code snippet is straightforward, and it differs from the binary search approach originally suggested. Here's the thought process behind the implementation given the reference to use a binary search approach:

Binary Search: Since the arrays are sorted, for each element in arr1, we can perform a binary search in arr2 and arr3. If the element is found in both other arrays, then it's added to our result list. Binary search is preferred because it can drastically reduce the search time in a sorted array from linear (O(n)) to logarithmic (O(log n)).
Counter for Frequency: The actual code uses the Counter from Python's collections module. A Counter is a subclass of dict designed to count hashable objects. It's an unordered collection where elements are stored as dictionary keys and their counts as dictionary values.
Concatenation and Counting: The solution concatenates arr1, arr2, and arr3 using the + operator. Once concatenated, a Counter object is created which tallies the count of each unique element across the three arrays.
Filtering Common Elements: We now have the frequency of each number that occurs in the three arrays. Since we're interested in numbers that appear in all three arrays, we filter out the elements which have a frequency count of 3. The code snippet does this using a list comprehension [x for x in arr1 if cnt[x] == 3]. Here, cnt[x] retrieves the count for element x from the Counter object.
Sorted Output: The given code doesn't explicitly sort the output list because it's already guaranteed to be sorted. This is because we are iterating through arr1 (which is sorted) and we're only picking those elements that are common to all arrays, hence maintaining the original order.

The binary search approach (not implemented in the solution code), if exampled, would require iterating through arr1 and for each element 'x', perform binary search on arr2 and arr3. If 'x' is found in both these arrays, it indicates that 'x' is common across all three arrays and should be added to the result.

However, the provided solution is likely more efficient since:

Counting elements across all arrays is done in linear time relative to the total number of elements.
No additional logarithmic factor is introduced as it would be with binary searches.
It makes use of the problem constraint (no duplicates within the same array) to simplify the solution.

Ready to land your dream job?

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

Start Evaluator

Example Walkthrough

To illustrate the solution approach, consider the following small example:

Let arr1 = [1,2,3], arr2 = [1,3,5], and arr3 = [1,3,7].

Now, let's step through the mixed solution approach using a frequency counter as implemented in the code snippet (despite binary search being advised):

Concatenation and Counting: We start by concatenating arr1, arr2, and arr3 to get [1,2,3,1,3,5,1,3,7].
Using Counter: A Counter object is created, and it counts the frequency of each element in the combined list: {1: 3, 2: 1, 3: 3, 5: 1, 7: 1}.
Filtering Common Elements: Given the frequency count, we're interested in elements with a count of 3, indicating they are present in all three arrays. We iterate through arr1 and use a list comprehension to find the common elements [x for x in arr1 if cnt[x] == 3], resulting in [1,3], since those elements have a frequency of 3.
Sorted Output: The final result is [1,3], which is the list of common integers in all three arrays.

The advantage of this solution is that it operates in linear time relative to the sum of elements in all arrays without needing to sort since the arrays are already sorted. Furthermore, it avoids the potentially more complex implementation and additional search time complexity of using binary search for each element.

Solution Implementation

1from typing import List
2from collections import Counter
3
4class Solution:
5    def arrays_intersection(self, arr1: List[int], arr2: List[int], arr3: List[int]) -> List[int]:
6        # Create a counter object which will store the frequency of each number across all three arrays
7        elements_count = Counter(arr1 + arr2 + arr3)
8      
9        # Find the intersection by checking which numbers have a count equal to 3 - meaning they appear in all three arrays
10        intersection = [number for number in arr1 if elements_count[number] == 3]
11      
12        # Return the list of numbers that are present in all three arrays
13        return intersection
14

1class Solution {
2  
3    // Method to find the intersection of three sorted arrays
4    public List<Integer> arraysIntersection(int[] arr1, int[] arr2, int[] arr3) {
5        // List to store the common elements across the three arrays
6        List<Integer> intersection = new ArrayList<>();
7        // Counter array to store the frequency of elements (assuming all elements are <= 2000)
8        int[] count = new int[2001];
9      
10        // Increment the counter for every element in arr1
11        for (int num : arr1) {
12            count[num]++;
13        }
14      
15        // Increment the counter for every element in arr2
16        for (int num : arr2) {
17            count[num]++;
18        }
19      
20        // Check arr3 and add the number to the result list if it occurs in all three arrays
21        for (int num : arr3) {
22            // If the count reaches 3, it means the element is present in all three arrays
23            if (++count[num] == 3) {
24                intersection.add(num);
25            }
26        }
27      
28        // Return the list of common elements
29        return intersection;
30    }
31  
32}
33

1#include <vector>
2
3class Solution {
4public:
5    // Function to find the intersection of three sorted arrays.
6    // The intersection contains elements that appear in all three arrays.
7    std::vector<int> arraysIntersection(std::vector<int>& arr1, std::vector<int>& arr2, std::vector<int>& arr3) {
8        // Create a result vector to store the common elements.
9        std::vector<int> result;
10
11        // Initialize a count array to store the frequency of each element.
12        // The problem statement ensures no element is greater than 2000.
13        int count[2001] = {}; // Initialize all counts to zero.
14
15        // Increment the count for every element in arr1
16        for (int elem : arr1) {
17            ++count[elem];
18        }
19
20        // Increment the count for every element in arr2
21        for (int elem : arr2) {
22            ++count[elem];
23        }
24
25        // Checking the elements of arr3;
26        // if the count becomes 3, it means the element is common to arr1, arr2, and arr3.
27        for (int elem : arr3) {
28            if (++count[elem] == 3) {
29                result.push_back(elem); // Add the common element to the result vector.
30            }
31        }
32
33        // Return the vector containing the intersecting elements.
34        return result;
35    }
36};
37

1// Declare three arrays to hold the sorted input arrays.
2let arr1: number[];
3let arr2: number[];
4let arr3: number[];
5
6// Function to find the intersection of three sorted arrays.
7// The intersection contains elements that appear in all three arrays.
8function arraysIntersection(arr1: number[], arr2: number[], arr3: number[]): number[] {
9    // Create an array to store the result of the intersection.
10    let result: number[] = [];
11
12    // Create an array to count the occurrences of each element.
13    // The problem statement ensures no element is greater than 2000,
14    // so we initialize an array of length 2001 to count all possible elements.
15    let count: number[] = new Array(2001).fill(0);
16
17    // Increment the count for each element in arr1.
18    arr1.forEach(elem => {
19        count[elem]++;
20    });
21
22    // Increment the count for each element in arr2.
23    arr2.forEach(elem => {
24        count[elem]++;
25    });
26
27    // Loop through the elements in arr3 and check if the count is 2 before the loop,
28    // If it becomes 3 after incrementing, that means the element
29    // is common to arr1, arr2, and arr3.
30    arr3.forEach(elem => {
31        if (++count[elem] === 3) {
32            result.push(elem); // Add the common element to the result array.
33        }
34    });
35
36    // Return the array containing the intersecting elements.
37    return result;
38}
39
40// Example usage:
41// arr1 = [1, 2, 3];
42// arr2 = [1, 3, 4];
43// arr3 = [1, 3, 5];
44// const intersection = arraysIntersection(arr1, arr2, arr3);
45// console.log(intersection); // Should log [1, 3] to the console.
46

Time and Space Complexity

Time Complexity

The time complexity of the code is O(n + m + l), where n is the length of arr1, m is the length of arr2, and l is the length of arr3. This is because the code concatenates all three arrays, which takes O(n + m + l) time, and then constructs a Counter object from the concatenated array, which also takes O(n + m + l) time as each element is processed once. Finally, a list comprehension is used which goes through each element in arr1 (O(n) time) and checks the count, which is an O(1) operation due to the hash table lookup in Counter. Therefore, the overall time complexity is dominated by the concatenation and the Counter operation.

Space Complexity

The space complexity of the code is O(n + m + l) because we are creating a Counter object that contains all the elements from arr1, arr2, and arr3. This requires space proportional to the total number of elements in all three arrays. The list comprehension does not add additional space as it only contains the elements that are common in all three arrays (at most O(n)), but this does not change the overall space complexity dominated by the Counter.

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

Which type of traversal does breadth first search do?

Pre-order traversal

Post-order traversal

In-order traversal

Level-order traversal

I don’t know