问题描述:
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]输出: [2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]输出: [9,4]
说明:
- 输出结果中的每个元素一定是唯一的。
- 我们可以不考虑输出结果的顺序。
方法1:
1 class Solution(object): 2 def intersection(self, nums1, nums2): 3 """ 4 :type nums1: List[int] 5 :type nums2: List[int] 6 :rtype: List[int] 7 """ 8 s_list = [] 9 nums1 = set(nums1)10 nums2 = set(nums2)11 for i in nums1:12 if i in nums2:13 s_list.append(i)14 return s_list
官方:
1 class Solution(object):2 def intersection(self, nums1, nums2):3 """4 :type nums1: List[int]5 :type nums2: List[int]6 :rtype: List[int]7 """8 nums1 = set(nums1)9 return [x for x in set(nums2) if x in nums1]
官方2:
1 class Solution(object):2 def intersection(self, nums1, nums2):3 """4 :type nums1: List[int]5 :type nums2: List[int]6 :rtype: List[int]7 """8 return list(set(nums1) & set(nums2))
2018-09-26 14:55:19