需求:
给定两个数组,编写函数计算出它们的交集。
示例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 的大小比 nums2 小很多,哪种方法更优?
- 如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
废话少说直接上代码:
class Solution:
def intersect(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
nums1.sort()
nums2.sort()
new_li = list()
i, j = 0, 0
while i < len(nums1) and j < len(nums):
if nums1[i] == nums2[j]:
new_li.append(nums1[i])
i += 1
j += 1
elif nums1[i] < nums2[j]:
i += 1
else:
j += 1
return new_li
总结:
先创建用于存储相同元素的空列表,然后初始化游标,在游标不超出数组长度的情况下,判断两个数组里面的元素在当前索引位置是否相等,相等就添加进入新列表,同时移动两个游标(因为已判为相等,元素相当于被消耗掉了),如果不相等,则根据元素的大小移动较小一方的游标(因为数组已排序),知道任意索引超过数组长度跳出循环,返回新数组,即两数组的交集元素。