给定两个数组,写一个函数来计算它们的交集。
例子:
给定 num1= [1, 2, 2, 1]
, nums2 = [2, 2]
, 返回 [2]
.
提示:
- 每个在结果中的元素必定是唯一的。
- 我们可以不考虑输出结果的顺序。
class Solution(object):
def intersection(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
setNums1 = set(nums1)
setNums2 = set(nums2)
result = []
for x in setNums2:
if x in setNums1:
result.append(x)
return result
用set来把list的重复元素过滤掉,然后判断是否存在,把结果保存起来
http://www.waitingfy.com/archives/3724