【题目】
给定两个数组,编写一个函数来计算它们的交集。
【示例 1】
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
【示例 2】
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
【说明】
输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。
【代码】
【Python】
class Solution:
def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
num1_dict={}
rs=set()
for x in nums1:
num1_dict[x]=1
for x in nums2:
if x in num1_dict:
rs.add(x)
return rs
【方法2-set()集合】
执行用时:
52 ms, 在所有 Python3 提交中击败了86.56%的用户
内存消耗:
15.1 MB, 在所有 Python3 提交中击败了13.94%的用户
class Solution:
def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
return list(set.intersection(set(nums1),set(nums2)))
【方法3】
class Solution:
def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
set1,set2=set(nums1),set(nums2)
return [x for x in set1 if x in set2]