给定两个数组,写一个方法来判断它们的交集
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/