Intersection of two arrays

November 2, 2022

intervals

Problem URL: Intersection of two arrays

We can use 2 sets to store the elements of each array. Then we can iterate over the elements of the first set and check if it is present in the second set. If it is, we add it to the result.

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        s1 = set(nums1)
        s2 = set(nums2)

        result = []
        for num in s1:
            if num in s2:
                result.append(num)

        return result

Time complexity: O(n)
Space complexity: O(n)

We can use 2 sets and get the intersection of the 2 sets.

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        return set(nums1) & set(nums2)

Time complexity: O(n)
Space complexity: O(n)