LeetCode 350 [Intersection of Two Array II]

原題

計算兩個陣列的交
注意事項
每個元素出現次數得和在陣列裡一樣答案可以以任意順序給出

樣例
nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].

解題思路

• 遍歷第一個陣列，構建hash map，計數
• 遍歷第二個陣列，如果存在在hash map而且個數大於零，加入res陣列
• 其他方法：先排序，兩根指標

完整程式碼

class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        map = {}
        for x in nums1:
            if x not in map:
                map[x] = 1
            else:
                map[x] += 1
            
        res = []
        for y in nums2:
            if y in map and map[y] > 0:
                res.append(y)
                map[y] -= 1
            
        return res