349.intersection-of-two-arrays

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

例子:

 给定 num1[1, 2, 2, 1]nums2 = [2, 2], 返回 [2].

提示:

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

解题思路:

    遍历第一个数组,并存到一个集合中去,然后再遍历第二个数组,如果第二个数组中存在集合中的元素,则将其放入ans中,并删除集合中的这个元素。

解题代码:

    

        d={}
        ans=[]
        for num in nums1:
            d[num]=d.get(num,1)
        for num in nums2:
            if num in d:
                ans.append(num)
                del d[num]
        return ans

你可能感兴趣的:(python,Leetcode)