Leetcode350:两个数组的交集 II

题目描述:

给你两个整数数组 nums1nums2 ,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序。

代码思路:

  1. 参数检查与交换
    • 首先,方法检查两个列表的长度。如果nums2的长度小于nums1的长度,就交换这两个列表的角色,即调用self.intersect(nums2, nums1)。这样做的目的是为了优化性能,确保在后续步骤中处理的列表是较长的那个。因为使用Counter对较长的列表进行计数会更高效,同时在后续遍历较短的列表时,可以减少查找次数。
  2. 初始化结果列表和计数器
    • 定义一个空列表ans,用于存储最终的交集结果。
    • 使用Counter类(来自collections模块)对nums1进行计数,生成一个字典c,其中键是nums1中的元素,值是这些元素在nums1中出现的次数。
  3. 遍历较短列表并查找交集
    • 遍历nums2中的每个元素n
    • 如果n在计数器c中存在(即nnums1中的一个元素),则将n添加到结果列表ans中。
    • 然后,将计数器cn的计数减1,表示已经找到一个与nums2中的元素相匹配的nums1中的元素。
    • 如果减1后n的计数变为0(意味着nums1中的所有匹配n的元素都已经被使用完),则从计数器c中移除键n。这是为了优化后续查找的性能,因为不再需要查找已经用完的元素。
  4. 返回结果
    • 最后,返回结果列表ans,其中包含了nums1nums2的交集。

代码实现:

复制代码
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
相关推荐
朔北之忘 Clancy24 分钟前
第二章 分支结构程序设计(3)
c++·算法·青少年编程·竞赛·教材·考级·讲义
毅炼28 分钟前
hot100打卡——day09
java·leetcode
想逃离铁厂的老铁33 分钟前
Day42 >> 188、买卖股票的最佳时机IV + 309.最佳买卖股票时机含冷冻期 + 714.买卖股票的最佳时机含手续费
算法·leetcode·职场和发展
wu_asia35 分钟前
方阵对角线元素乘积计算
数据结构·算法
想逃离铁厂的老铁1 小时前
Day43 >> 300.最长递增子序列 + 674. 最长连续递增序列+ 718. 最长重复子数组
数据结构·算法
Yzzz-F1 小时前
P6648 [CCC 2019] Triangle: The Data Structure [st表]
算法
LateFrames1 小时前
泰勒级数:从 “单点” 到 “理论与实践的鸿沟”
学习·算法
武帝为此2 小时前
【RC4加密算法介绍】
网络·python·算法
宵时待雨2 小时前
数据结构(初阶)笔记归纳4:单链表的实现
c语言·开发语言·数据结构·笔记·算法
BLSxiaopanlaile2 小时前
关于子集和问题的几种解法
数据结构·算法·剪枝·回溯·分解