LeetCode--349--两个数组的交集

问题描述:

给定两个数组，编写一个函数来计算它们的交集。

示例 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