Leetcode350：两个数组的交集 II

m0_675988232025-02-24 14:18

题目描述：

给你两个整数数组 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