python(leetcode)-350两个数组的交集

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

示例 1:

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

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

输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
进阶:

如果给定的数组已经排好序呢？你将如何优化你的算法？
如果 nums1 的大小比 nums2 小很多，哪种方法更优？
如果 nums2 的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

看完问题描述 发现看似简单 但是其实有一点点的难 难点在于如果2个数组中存在重复数字，则交集也必须重复

先说下 自己当时的想法

没有想到什么简单方法 只能使用蛮力法 通过两层循环 每一个数字与另外一个数组中的数字对比。并且加一个判断，如果相同则在之后的对比时不对比此下标的值

上代码（通过-用时110ms)

 class Solution:

     def intersect2(self, nums1, nums2):
         """
         :type nums1: List[int]
         :type nums2: List[int]
         :rtype: List[int]
         """
         result = []

         for i in nums1:
             for j in nums2:
                 if i == j:
                     result.append(i) #插入交集
                     nums2.remove(j)  #插入过的值不再出现
                     break
         return result
 if __name__=="__main__":
     nums1=[1,2,2,1]
     nums2=[1,2]
     s=Solution()
     print(s.intersect(nums1,nums2))

这个是笔者想到的方法。使用蛮力法，比较耗时，需要110ms。

然后上评论区大佬的代码（特别赏心悦目 5行代码解决问题）

 class Solution(object):
     def intersect(self, nums1, nums2):
         """
         :type nums1: List[int]
         :type nums2: List[int]
         :rtype: List[int]
         """
         inter = set(nums1) & set(nums2)     #inter为两个list的相同元素
         l = []
         for i in inter:
             l += [i] * min(nums1.count(i), nums2.count(i)) #相同元素的共同出现次数
         return l

这个方法比较难想，是一个很有趣的解法

思想：通过set的唯一性找出两list的唯一交集，再利用min方法找到交集的共同出现次数

前面方法的效率较低 介绍一种方法使用字典进行查找

思想：通过key,value进行存储，key为值的个数，value为值。 先把nums1存放入字典中。遍历nums2中的值对比字典，相同则数量减一并放入list中，最后返回list

 class Solution:
     #利用字典实现交集
     def intersect(self,nums1,nums2):
         record={}
         result=[]
         for i in nums1:
             # 注意python3.x用contains()方法取代has_key()
             if(record.__contains__(i)):
                 record[i]+=1                      #将字典的key值数+1
             else:
                 record[i]=1                       #将i加入字典中 key为1
         for j in nums2:
             if(record.__contains__(j) and record[j]>0):
                 record[j]-=1                      #i的key值减1
                 result.append(j)
         return result
 if __name__=="__main__":
     s=Solution()
     num1=[1,2,3,4,4,1]
     num2=[4,2,3]
     print(s.intersect(num1,num2))

值得注意的是：python 3.x的版本将has_key()方法换为_contains_()

会发现通过时间为48ms 效率有所提升