思想:
题目要求给定两个数组,编写一个函数来计算它们的交集。
1.定义变量ans储存两个数组的交集
2.将nums1和nums2去重
3.判断nums1中的元素num是否在nums2中,若在则添加ans中,不在则继续for循环
class Solution(object):
def intersection(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
ans=[]
nums1=set(nums1)
nums2=set(nums2)
for num in nums1:
if num in nums2:
ans.append(num)
return ans