Intersection of Two Arrays

Given two arrays, write a function to compute their intersection.

Example: Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].

Note: Each element in the result must be unique. The result can be in any order.

class Solution(object):
    def intersection(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        res = []
        for n in nums1:
            if n in nums2 and n not in res:
                res += n,
        return res

A bit faster

class Solution(object):
    def intersection(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        res = {}
        val = []
        for n in nums1:
            res[n] = 1
        for n in nums2:
            if n in res:
                res[n] += 1
        for k,v in res.items():
            if v > 1:
                val += k,
        return val

results matching ""

    No results matching ""