题目要求
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
思路分析:
这里可以选择两种办法,一个是列表推导式法,一个直接用set函数可以解决。
代码实现:
class Solution(object):
def intersection(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
temp=[val for val in nums1 if val in nums2]#val 是要返回的值
return set(temp)#去重,因为有可能出现两个相同的元素
## 法二
return list(set(nums1)&set(nums2))