[leetcode]Python实现-349.两个数组的交集

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}