Skip to content

Latest commit

 

History

History
29 lines (27 loc) · 845 Bytes

350_intersection_of_two_arrays_2.md

File metadata and controls

29 lines (27 loc) · 845 Bytes

Given two integer arrays nums1 and nums2, return an array of their intersection. Each element in the result must appear as many times as it shows in both arrays and you may return the result in any order.

Example 1:

Input: nums1 = [1,2,2,1], nums2 = [2,2]
Output: [2,2]

Example 2:

Input: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
Output: [4,9]
Explanation: [9,4] is also accepted.

Solution

from collections import Counter
class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        nums1 = Counter(nums1)
        nums2 = Counter(nums2)
        out = []
        for n, c in nums1.items():
            if n in nums2:
                out.extend([n] * min(c, nums2[n]))
        return out