leetcode349 350 Intersection of Two Arrays & II

"""
Intersection of Two Arrays
Given two arrays, write a function to compute their intersection.
Example 1:
Input: nums1 = [1,2,2,1], nums2 = [2,2]
Output: [2]
Example 2:
Input: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
Output: [9,4]
"""
"""
由于问题中的元素是唯一的，所以我们只关心元素的有无，
那么我们可以使用set这个结构。
首先将nums1的所有数据存入set中，查找nums2中的数据是否在这个set中，
如果在的话，我们将这个元素存入一个list里面。
"""
class Solution:
    def intersection(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        nums1 = set(nums1)
        result = set()
        for i in nums2:
            if i in nums1:
                result.add(i)
        return list(result)

"""
Given two arrays, write a function to compute their intersection.
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]
"""
"""
发现在leetcode349这个问题的基础上，
我们多加了记录元素个数的功能，我们可以使用dict实现它。
"""
class Solution:
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        record, result = {}, []
        for num in nums1:
            record[num] = record.get(num, 0) + 1
            # dict.get(key, default=None)返回指定键的值，如果值不在字典中返回默认值None
            #{1: 2, 2: 2}
        for num in nums2:
            if num in record and record[num]:
            # num in record 里而且 对应有重复值
                result.append(num)
                record[num] -= 1
        return result