【数据结构和算法】【刷题】Day1 349. 两个数组的交集（排序，简单,python3）

最新推荐文章于 2022-09-26 16:41:41 发布

原创最新推荐文章于 2022-09-26 16:41:41 发布 · 226 阅读

0 ·

CC 4.0 BY-SA版权

数据结构同时被 2 个专栏收录

6 篇文章

订阅专栏

leetcode

4 篇文章

订阅专栏

这篇博客介绍了如何计算两个数组的交集，使用Python3实现。通过创建nums1的集合并遍历nums2，检查元素是否在集合中，将唯一元素添加到结果列表。讨论了集合操作的效率，并引用了其他排序相关的解法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.题目：

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

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]
示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]
说明:

输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/intersection-of-two-arrays

------------------------------------------------------------------------------------------------------

2.解题

2.1

结合b站上 Michelle小姐姐的思路，

这道题目中先将nums1创建集合，提取其唯一的元素，再遍历nums2，判断nums2中的元素是否在nums1的集合中，若有，即有重复元素，在ans列表上加入这个元素，再在nums1的集合中删除这个元素，因为题目要求结果的元素是唯一的。

代码如下：

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        ans = []
        num1 = set(nums1)
        for i in nums2:
            if i  in num1:
                ans.append(i)
                num1.remove(i)
        return ans

并没有对原视频代码优化，只是结合目前本人的理解简单修改。用到了我不太熟悉的set，set中的元素是没有重复的，无序的

形如a={'a','b','c','d'}。

创建集合：a={'a','b','c','d'}或 a=set('abcd')

增加元素： add(),删除元素remove（）

**Python set集合操作**
数学符号	Python符号	含义
- 或\	-	差集，相对补集
∩	&	交集
∪	\|	并集
≠	!=	不等于
＝	==	等于
∈	in	是成员关系
∉	not in	非成员关系

根据集合求交集的操作&，可以有如下做法，不过时间和空间都没有优化。

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        num1 = set(nums1)
        num2 = set(nums2)
        return list(num1 & num2)

------------------------------------------------------------------------------------------------------

2.2

该题目的分类是排序，但是以上做法似乎与排序没有关系。

引用一下其他人的解法：

leetcode 专题：排序 python 版（持续更新中）【作者：Yenpo_Ma 】

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        if not nums1 or not nums2:
            return []
        nums1.sort()
        nums2.sort()
        len1 = len(nums1)
        len2 = len(nums2)
        result = []
        while nums1 and nums2:
            if nums1[0] == nums2[0]:
                if not nums1[0] in result: 
                    result.append(nums1[0])
                nums1.pop(0)
                nums2.pop(0)
            elif nums1[0]>nums2[0]:
                nums2.pop(0)
            elif nums1[0] < nums2[0]:
                nums1.pop(0)
        return result