LeetCode-350 两个数组的交集 II
2019-02-23 本文已影响0人
FlyCharles
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
在列表末尾添加n
个x
-
list.count(x)
列表统计元素x
的个数