Leetcode练习(Python):第349题:两个数组的交集:给定两个数组,编写一个函数来计算它们的交集。

题目:

两个数组的交集:给定两个数组,编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]
示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]
说明:

输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。

思路:

看到要求不重复结果使用哈希表。

程序:

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        if not nums1 and nums2:
            return []
        if nums1 and not nums2:
            return []
        if not nums1 and not nums2:
            return []
        result = set()
        for index in range(len(nums1)):
            if nums1[index] in nums2:
                result.add(nums1[index])
        return result