- 题目描述
给定两个数组,编写一个函数来计算它们的交集。
- 示例
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2] 输出: [2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出: [9,4]
说明:
- 输出结果中的每个元素一定是唯一的。
- 我们可以不考虑输出结果的顺序。
- 解题思路
我只想到了可以用python的set来对数组去重,查了资料发现了python的 Set intersection() 方法,用于返回两个或更多集合中都包含的元素,即交集。用法为:
set.intersection(set1, set2 ... etc)
set1 -- 必需,要查找相同元素的集合;set2 -- 可选,其他要查找相同元素的集合,可以多个,多个使用逗号 , 隔开;返回值是一个新的集合set。
- 拓展
与之类似的set的方法有:union求并集,difference求差集
list(set(a).union(set(b)))
list(set(b).difference(set(a)))
- 代码
class Solution(object):
def intersection(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
return list(set(nums1).intersection(set(nums2)))