Python实现"两个数组的交集"的两种方法

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/qiubingcsdn/article/details/82947880

给定两个数组,写一个方法来判断它们的交集

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]

注意:

输出结果中每个元素值唯一

输出结果中元素可随便排列

1:list()方法 + set()方法

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

另一种写法(参考他人)

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

2:和第一种方法类似

def intersection(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        nums2 = set(nums2)
        return [i for i in set(nums1) if i in nums2]

去掉语句"nums2 = set(nums2)",导致运行时间大大增加(从28ms - 104ms)。说明set()语句执行需要消耗大量的时间

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

算法题来自:https://leetcode-cn.com/problems/intersection-of-two-arrays/description/

展开阅读全文

没有更多推荐了,返回首页