349.两个数组的交集
描述
给定两个数组,写一个函数来计算它们的交集。
例子
给定 num1= [1, 2, 2, 1], nums2 = [2, 2], 返回 [2].
提示
每个在结果中的元素必定是唯一的。
我们可以不考虑输出结果的顺序。
我
class Solution:
def intersection(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
l1 = len(nums1)
l2 = len(nums2)
res = set()
if l1 <= l2:
s = nums1
l = nums2
else:
l = nums1
s = nums2
for i in l:
if i in s:
res.add(i)
return list(res)
别人的
class Solution:
def intersection(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
s1, s2 = set(nums1), set(nums2)
return list(s1.intersection(s2))
关于Python的set有一个intersection()方法。
intersection(...)
| Return the intersection of two sets as a new set.
|
| (i.e. all elements that are in both sets.)
求交集、并集和差集
>>> set.intersection(s,d)#获取s,d的交集
{1}
>>> s = {1,2,3,4,5}
>>> d = {2,3,4,5,6}
>>> s.intersection(d)
{2, 3, 4, 5}
>>> s.union(d)#并集
{1, 2, 3, 4, 5, 6}
>>> s.difference(d)#获取差集s-d
{1}
>>> d.difference(s)#获取差集d-s
{6}