python leetcode 349. 两个数组的交集

给定两个数组，编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]

示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]

说明:

输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。

---

这个工作中一定是用现成的函数和方法,简单粗暴。

class Solution:
    def intersection(self, nums1, nums2):     
        s1, s2 = set(nums1), set(nums2)
        return list(s1.intersection(s2))

面试的话有可能，不允许用现成的。那么看看下面这个

class Solution:
    def intersection(self, nums1, nums2):
        nums1 =set(nums1)
        nums2 =set(nums2)
        res = []
        if len(nums1) > len(nums2):
            for i in nums2:
                if i in nums1 and i not in res:
                    res.append(i)
        else:
            for i in nums1:
                if i in nums2 and i not in res:
                    res.append(i)
        return res

逻辑还是简单的，就是取合集去重，然后遍历长度小一些的列表元素，在长列表中查找，找到保存。