1. 題目
https://leetcode-cn.com/problems/intersection-of-two-arrays/
給定兩個數(shù)組，編寫一個函數(shù)來計算它們的交集籍茧。
示例 1:
輸入: nums1 = [1,2,2,1], nums2 = [2,2]
輸出: [2]
示例 2:
輸入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
輸出: [9,4]
說明:
輸出結(jié)果中的每個元素一定是唯一的喷舀。
我們可以不考慮輸出結(jié)果的順序。
2. 我的AC
集合函數(shù)set
方法1
class Solution(object):
def intersection(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
return list(set(nums1) & set(nums2))
方法2
遍歷數(shù)組
class Solution(object):
def intersection(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
result = []
for num1 in nums1:
if num1 not in result and num1 in nums2:
result.append(num1)
return result
3. 小結(jié)
- 兩個數(shù)組的交集、并集
list(set(nums1) & set(nums2)) # 交集
list(set(nums1) | set(nums2)) # 并集
