给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2,2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
# 双指针分别指向排序后的nums1、nums2,判断其所指元素是否相等,若不等则向右移位
class Solution:
def intersect(self, nums1: [int], nums2: [int]) -> [int]:
nums1.sort()
nums2.sort()
res = []
left, right = 0, 0
while left < len(nums1) and right < len(nums2):
if nums1[left] == nums2[right]:
res.append(nums1[left])
left += 1
right += 1
elif nums1[left] < nums2[right]:
left += 1
else:
right += 1
return res
# 使其长度较短的为nums1,用i遍历,若i也存在于nums2,将i添加到最终结果的列表并在nums2中删除i(速度较慢)
class Solution:
def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
if len(nums1) > len(nums2):
nums1, nums2 = nums2, nums1
res = []
for i in nums1:
if i in nums2:
res.append(i)
nums2.remove(i)
return res