LeetCode349. Intersection of Two Arrays

船桥

于 2018-12-11 23:04:12 发布

阅读量104

点赞数

CC 4.0 BY-SA版权

分类专栏： Python LeetCode

本文链接：https://blog.youkuaiyun.com/AIpush/article/details/84963305

LeetCode 同时被 2 个专栏收录

179 篇文章

订阅专栏

Python

173 篇文章

订阅专栏

本文介绍了一种计算两个数组交集的算法实现，通过将数组转换为集合并使用Python内置的intersection方法，有效地找到两个数组中的共同元素。最终返回一个包含唯一交集元素的列表。

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

Given two arrays, write a function to compute their intersection.

Example 1:

Input: nums1 = [1,2,2,1], nums2 = [2,2]
Output: [2]

Example 2:

Input: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
Output: [9,4]

Note:

Each element in the result must be unique.
The result can be in any order.

思路：求两个列表的交集，就是在两个列表中同时存在。最后需要用list()表达出来。

class Solution:
    def intersection(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        return list(set([val for val in nums1 if val in nums2]))

也可以用python自带的intersection()方法

class Solution:
    def intersection(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        return list(set(nums1).intersection(set(nums2)))