题目描述:
给你两个整数数组 nums1
和 nums2
,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序。
代码思路:
- 参数检查与交换 :
- 首先,方法检查两个列表的长度。如果
nums2
的长度小于nums1
的长度,就交换这两个列表的角色,即调用self.intersect(nums2, nums1)
。这样做的目的是为了优化性能,确保在后续步骤中处理的列表是较长的那个。因为使用Counter
对较长的列表进行计数会更高效,同时在后续遍历较短的列表时,可以减少查找次数。
- 首先,方法检查两个列表的长度。如果
- 初始化结果列表和计数器 :
- 定义一个空列表
ans
,用于存储最终的交集结果。 - 使用
Counter
类(来自collections
模块)对nums1
进行计数,生成一个字典c
,其中键是nums1
中的元素,值是这些元素在nums1
中出现的次数。
- 定义一个空列表
- 遍历较短列表并查找交集 :
- 遍历
nums2
中的每个元素n
。 - 如果
n
在计数器c
中存在(即n
是nums1
中的一个元素),则将n
添加到结果列表ans
中。 - 然后,将计数器
c
中n
的计数减1,表示已经找到一个与nums2
中的元素相匹配的nums1
中的元素。 - 如果减1后
n
的计数变为0(意味着nums1
中的所有匹配n
的元素都已经被使用完),则从计数器c
中移除键n
。这是为了优化后续查找的性能,因为不再需要查找已经用完的元素。
- 遍历
- 返回结果 :
- 最后,返回结果列表
ans
,其中包含了nums1
和nums2
的交集。
- 最后,返回结果列表
代码实现:
from collections import Counter
class Solution:
def intersect(self, nums1: list[int], nums2: list[int]) -> list[int]:
if len(nums2) < len(nums1):
return self.intersect(nums2, nums1)
ans = []
c = Counter(nums1)
for n in nums2:
if n in c:
ans.append(n)
c[n] -= 1
if c[n] == 0:
c.pop(n)
return ans