問題描述:

給定兩個數組，編寫一個函數來計算它們的交集。

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

LeetCode--349--兩個數組的交集