美文网首页
LeetCode-350 两个数组的交集 II

LeetCode-350 两个数组的交集 II

作者: FlyCharles | 来源:发表于2019-02-23 14:05 被阅读0次

1. 题目

https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/

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

示例 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. 我的AC

class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        result = []
        for num1 in nums1:
            if num1 in nums2:
                result.append(num1)
                nums2.remove(num1) # Important,注意以下测试案例
        return result

测试案例

输入
[1,2,2,1]
[2]
输出
[2,2]
预期结果
[2]

思路2

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

class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        inter = set(nums1) & set(nums2)
        result = []
        for n in inter:
            result += [n] * min(nums1.count(n), nums2.count(n))  
        return result

3. 小结

  • list.append(x) 在列表末尾添加元素
  • list += [x] 在列表末尾添加元素
  • list += [x] * n 在列表末尾添加nx
  • list.count(x) 列表统计元素x的个数

相关文章

网友评论

      本文标题:LeetCode-350 两个数组的交集 II

      本文链接:https://www.haomeiwen.com/subject/bsiuyqtx.html