题目描述
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
思路分析
- 从两者长度小的那一方着手,如果nums1长度较小,则遍历nums1,将nums1和nums2中的共同元素append进result,然后将该元素从nums2中remove掉,这一步是和349题不同的地方。
代码示例
class Solution(object):
def intersect(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
result = []
if len(nums1) <= len(nums2):
for num in nums1:
if num in nums2:
result.append(num)
nums2.remove(num)
return result
else:
for num in nums2:
if num in nums1:
result.append(num)
nums1.remove(num)
return result